jax.numpy.angle#

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

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

JAX 对 numpy.angle() 的实现。

参数:

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

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

返回:

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

返回类型:

Array

示例

如果 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]]