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.
x = tf.constant([-2.25 + 4.75j, 3.25 + 5.75j])
tf.math.imag(x) # [4.75, 5.75]
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.