tf.math.imag

View source on GitHub

Returns the imaginary part of a complex (or real) tensor.

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

Given a tensor input, this operation returns a tensor of type float that is the imaginary part of each element in input considered as a complex number. If input is real, a tensor of all zeros is returned.

For example:

x = tf.constant([-2.25 + 4.75j, 3.25 + 5.75j])
tf.math.imag(x)  # [4.75, 5.75]

Args:

Returns:

A Tensor of type float32 or float64.