jax.numpy.cos

jax.numpy.cos(x, /)

计算输入中每个元素的三角余弦。

numpy.cos 的 JAX 实现。

参数：

x (ArrayLike) – 标量或数组。以弧度为单位的角度。

返回值：

一个包含 x 中每个元素的余弦值的数组，提升为非精确数据类型。

返回类型：

数组

另请参阅

• 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]