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