如果您有一个复数张量,则tensorflow已经可以选择获取那些复数(tf.abs)的大小。现在,我想添加一个函数来计算每个数字的角度。使用numpy可以轻松实现此功能,并且渐变为known。
在添加基于numpy的自定义tensorflow操作时,this似乎是可行的方法。但是,这里的输入是复数。所以我的问题是,是否可以使用相同的方法但可以处理复数?由于直接使用tf.abs中的方法使用c ++,因此无法直接复制。
最佳答案
看起来tf.arg
即将存在:https://github.com/tensorflow/tensorflow/pull/10643。如果您在此之前想要一些东西,我建议
def arg(z):
return tf.atan2(tf.imag(z), tf.real(z))
关于python - 如何为在TensorFlow中处理复杂张量的自定义操作定义梯度?,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/44564397/