tf.math.angle

Возвращает аргумент каждого элемента комплексного (или вещественного) тензора.

tf.math.angle(
    input, name=None
)

Для данного тензора input, данная операция возвращает тензор типа float, содержащий аргумент каждого элемента в input , рассматриваемого как комплексное число.

Элементы в input рассматриваются как комплексные числа вида \(a + bj\), где a - действительная часть, а b - мнимая часть. Если input вещественный, то b по определению равно нулю.

Аргумент, возвращаемый этой функцией, имеет вид \(atan2(b, a)\). Если input вещественный, возвращается тензор из нулей.

Пример:

input = tf.constant([-2.25 + 4.75j, 3.25 + 5.75j], dtype=tf.complex64)
tf.math.angle(input).numpy()
# ==> array([2.0131705, 1.056345 ], dtype=float32)