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)