jax.numpy.angle#

jax.numpy.angle(z, deg=False)[源代码]#

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

JAX 实现的 numpy.angle()。

参数:

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

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

返回值:

一个由 z 中每个元素的逆时针角度组成的数组，其形状与 z 相同，dtype 为 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]]