Python | tensorflow.math.angle() 方法

tensorflow.math.angle()方法用于计算复数张量中每个元素的角度。角度表示为弧度，返回的张量具有与输入张量相同的尺寸和类型。

语法

tensorflow.math.angle(input, name=None)

参数

• input：一个类型是tf.float16, tf.float32, tf.float64, tf.complex64, tf.complex128的张量。表示输入数据。

• name：（可选）该操作的名称。

返回值

• 一个与输入相同尺寸和类型的张量，表示每个元素的角度。

示例代码

import tensorflow as tf

# 定义一个复数张量
input_tensor = tf.constant([1 + 1j, 2 + 2j], dtype=tf.complex64)

# 计算张量的角度
output_tensor = tf.math.angle(input_tensor)

# 输出结果
print(output_tensor.numpy())

输出结果为：

[0.7853982 0.7853982]

异常

• TypeError ：如果 input 的数据类型不是复数类型（tf.complex64 或者 tf.complex128 ），则抛出 TypeError 异常。

注意事项

• 由于角度用弧度表示，因此输出结果的范围为 $(-\pi, \pi]$ 。

• 输出的张量的类型与输入的张量的类型相同。如果输入是浮点数类型，则输出也是浮点数类型，如果输入是复数类型，则输出也是复数类型。

• tensorflow.math.angle()方法不支持自动广播，因此输入的张量和输出的张量必须具有相同的形状。