/TensorFlow 2.4


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

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)
# ==> array([2.0131705, 1.056345 ], dtype=float32)
input A Tensor. Must be one of the following types: float, double, complex64, complex128.
name A name for the operation (optional).
A Tensor of type float32 or float64.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.