jax.numpy.cos

jax.numpy.cos(x, /)

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

JAX implementation of numpy.cos.

参数:

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

返回:

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

返回类型:

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]