Python tf.math.angle用法及代码示例

返回复数(或实数)张量的元素参数。

用法

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

参数

input 一个Tensor。必须是以下类型之一：float、double、complex64、complex128。
name 操作的名称(可选)。

Tensor 类型为 float32 或 float64 。

给定一个张量 input ，此操作返回一个类型为 float 的张量，它是 input 中每个元素的参数，被视为复数。

中的元素input被认为是形式的复数\(a + bj\)，其中a是实部并且b是虚部。如果input那么是真的b根据定义为零。

此函数返回的参数格式为 \(atan2(b, a)\) 。如果input 为实数，则返回全零张量。

例如：

input = tf.constant([-2.25 + 4.75j, 3.25 + 5.75j], dtype=tf.complex64)
tf.math.angle(input).numpy()
# ==> array([2.0131705, 1.056345 ], dtype=float32)

相关用法

注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.math.angle。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。