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]]