gpt4 book ai didi

python - 如何为在 tensorflow 中处理复杂张量的自定义操作定义梯度?

转载 作者:行者123 更新时间:2023-12-01 02:52:41 26 4
gpt4 key购买 nike

如果你有一个复数张量,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/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com