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)