Функция complex
Функция complex создает комплексный тензор из
двух вещественных тензоров. Первым параметром
функция принимает действительную часть,
вторым - мнимую часть. Результатом является
тензор с типом данных complex64 или
complex128 в зависимости от типа входных
данных.
Синтаксис
tf.complex(real, imag, [name])
Пример
Давайте создадим комплексный тензор из
действительной части 1, 2, 3
и мнимой части 4, 5, 6:
import tensorflow as tf
t = tf.complex([1, 2, 3], [4, 5, 6])
print(t)
Результат выполнения кода:
tf.Tensor([1.+4.j 2.+5.j 3.+6.j], shape=(3,), dtype=complex128)
Пример
Давайте создадим комплексный тензор с типом
complex64, используя вещественные тензоры
типа float32:
<+python+>
import tensorflow as tf
real = tf.constant([1, 2, 3], dtype=tf.float32)
imag = tf.constant([4, 5, 6], dtype=tf.float32)
t = tf.complex(real, imag)
print(t)
<-python+>
Результат выполнения кода:
<+python+>
tf.Tensor([1.+4.j 2.+5.j 3.+6.j], shape=(3,), dtype=complex64)
<-python+>
Пример
Давайте создадим комплексный тензор из двумерных вещественных тензоров:
<+python+>
import tensorflow as tf
real = tf.constant([[1, 2, 3], [4, 5, 6]])
imag = tf.constant([[7, 8, 9], [10, 11, 12]])
t = tf.complex(real, imag)
print(t)
<-python+>
Результат выполнения кода:
<+python+>
tf.Tensor(
[[ 1. +7.j 2. +8.j 3. +9.j]
[ 4.+10.j 5.+11.j 6.+12.j]], shape=(2, 3), dtype=complex128)
<-python+>
Пример
Давайте извлечем действительную и мнимую части из комплексного тензора:
<+python+>
import tensorflow as tf
t = tf.complex([1, 2, 3], [4, 5, 6])
real_part = tf.math.real(t)
imag_part = tf.math.imag(t)
print(real_part)
print(imag_part)
<-python+>
Результат выполнения кода:
<+python+>
tf.Tensor([1. 2. 3.], shape=(3,), dtype=float64)
tf.Tensor([4. 5. 6.], shape=(3,), dtype=float64)
<-python+>
