tf.math.angle

TensorFlow 2 version

View source on GitHub

Returns the element-wise argument of a complex (or real) tensor.

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

Given a tensor input, this operation returns a tensor of type float that is the argument of each element in input considered as a complex number.

The elements in input are considered to be complex numbers of the form \(a + bj\), where a is the real part and b is the imaginary part. If input is real then b is zero by definition.

The argument returned by this function is of the form \(atan2(b, a)\). If input is real, a tensor of all zeros is returned.

For example:

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)