tf.raw_ops.ComplexAbs

Вычисляет модуль комплексного числа для тензора.

tf.raw_ops.ComplexAbs(
    x,
    Tout=tf.dtypes.float32,
    name=None
)

Для тензора x с комплексными числами эта операция возвращает тензор типа float или double, содержащий модуль каждого элемента в x. Все элементы в x должны быть комплексными числами вида \(a + bj\). Модуль вычисляется как \( \sqrt{a^2 + b^2}\).

Например:

x = tf.complex(3.0, 4.0)
print((tf.raw_ops.ComplexAbs(x=x, Tout=tf.dtypes.float32, name=None)).numpy())
5.0