tf.dtypes.complex

View source on GitHub

Converts two real numbers to a complex number.

View aliases

Main aliases

tf.complex

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.complex, tf.compat.v1.dtypes.complex, `tf.compat.v2.complex`, `tf.compat.v2.dtypes.complex`

tf.dtypes.complex(
    real, imag, name=None
)

Given a tensor real representing the real part of a complex number, and a tensor imag representing the imaginary part of a complex number, this operation returns complex numbers elementwise of the form \(a + bj\), where a represents the real part and b represents the imag part.

The input tensors real and imag must have the same shape.

For example:

real = tf.constant([2.25, 3.25])
imag = tf.constant([4.75, 5.75])
tf.complex(real, imag)  # [[2.25 + 4.75j], [3.25 + 5.75j]]

Args
real	A Tensor. Must be one of the following types: float32, float64.
imag	A Tensor. Must have the same type as real.
name	A name for the operation (optional).

Returns
A Tensor of type complex64 or complex128.

Raises
TypeError	Real and imag must be correct types