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]
Args | |
---|---|
input | A Tensor . Must be one of the following types: float , double , complex64 , complex128 . |
name | A name for the operation (optional). |
Returns | |
---|---|
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.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/math/imag