jax.numpy.cos

jax.numpy.cos(x, /)

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

JAX 对 numpy.cos 的实现。

参数:

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]