Функция imag
Функция imag возвращает мнимую часть комплексного тензора.
Первым параметром функция принимает комплексный тензор.
Если переданный тензор не является комплексным, функция возвращает
тензор, заполненный нулями, той же формы, что и входной.
Синтаксис
tf.math.imag(input, [name])
Пример
Давайте создадим комплексный тензор и извлечем его мнимую часть:
import tensorflow as tf
t = tf.constant([1 + 2j, 3 + 4j, 5 + 6j])
res = tf.math.imag(t)
print(res)
Результат выполнения кода:
tf.Tensor([2. 4. 6.], shape=(3,), dtype=float32)
Пример
Давайте применим функцию imag к вещественному тензору:
import tensorflow as tf
t = tf.constant([1, 2, 3, 4, 5])
res = tf.math.imag(t)
print(res)
Результат выполнения кода:
tf.Tensor([0. 0. 0. 0. 0.], shape=(5,), dtype=float32)
Пример
Давайте извлечем мнимую часть из двумерного комплексного тензора:
import tensorflow as tf
t = tf.constant([[1 + 2j, 3 + 4j], [5 + 6j, 7 + 8j]])
res = tf.math.imag(t)
print(res)
Результат выполнения кода:
tf.Tensor(
[[2. 4.]
[6. 8.]], shape=(2, 2), dtype=float32)
