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