Python tf.dtypes.complex用法及代码示例

将两个实数转换为复数。

用法

tf.dtypes.complex(
    real, imag, name=None
)

给定一个张量real表示复数的实部和张量imag表示复数的虚部，此操作返回形式为元素的复数\(a + bj\)，其中a代表real部分和b代表imag部分。

输入张量 real 和 imag 必须具有相同的形状。

例如：

real = tf.constant([2.25, 3.25])
imag = tf.constant([4.75, 5.75])
tf.complex(real, imag)  # [[2.25 + 4.75j], [3.25 + 5.75j]]