jax.numpy.angle#

jax.numpy.angle(z, deg=False)[原始碼]#

返回複數值數字或陣列的角度。

JAX 實現的 numpy.angle()。

參數：

  • z (ArrayLike) – 一個複數或複數陣列。

  • deg (bool) – 布林值。如果 True，則以度為單位返回結果，否則以弧度為單位返回。預設為 False。

返回：

一個 z 每個元素的逆時針角度陣列，形狀與 z 相同，dtype 為 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]]