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]