jax.numpy.angle

内容

jax.numpy.angle#

jax.numpy.angle(z, deg=False)[source]#

返回复数或复数数组的角度。

JAX 实现 numpy.angle()。

参数：

  • z (ArrayLike) – 复数或复数数组。

  • deg (bool) – 布尔值。如果为 True，则以度为单位返回结果，否则以弧度为单位返回结果。默认为 False。

返回值：

z 中每个元素的逆时针角的数组，形状与 z 相同，数据类型为 float。

返回类型：

数组

示例

如果 z 是一个数字

>>> z1 = 2+3j
>>> jnp.angle(z1)
Array(0.98279375, dtype=float32, weak_type=True)

如果 z 是一个数组

>>> z2 = jnp.array([[1+3j, 2-5j],
...                 [4-3j, 3+2j]])
>>> with jnp.printoptions(precision=2, suppress=True):
...     print(jnp.angle(z2))
[[ 1.25 -1.19]
 [-0.64  0.59]]

如果 deg=True。

>>> with jnp.printoptions(precision=2, suppress=True):
...     print(jnp.angle(z2, deg=True))
[[ 71.57 -68.2 ]
 [-36.87  33.69]]
内容