jax.numpy.angle

目录

jax.numpy.angle#

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

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

JAX 实现的 numpy.angle()。

参数:

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

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

返回:

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

返回类型:

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]]
目录