jax.numpy.cos

jax.numpy.cos(x, /)

計算輸入中每個元素的三角餘弦值。

numpy.cos 的 JAX 實作。

參數:

x (ArrayLike) – 純量或陣列。角度以弧度為單位。

傳回:

一個包含 x 中每個元素的餘弦值的陣列，提升為非精確 dtype。

傳回類型:

Array

另請參閱

• jax.numpy.sin()：計算輸入中每個元素的三角正弦值。

• jax.numpy.tan()：計算輸入中每個元素的三角正切值。

• jax.numpy.arccos() 和 jax.numpy.acos()：計算輸入中每個元素的三角餘弦反函數。

範例

>>> pi = jnp.pi
>>> x = jnp.array([pi/4, pi/2, 3*pi/4, 5*pi/6])
>>> with jnp.printoptions(precision=3, suppress=True):
...   print(jnp.cos(x))
[ 0.707 -0.    -0.707 -0.866]