jax.numpy.cos

jax.numpy.cos(x, /)

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

JAX实现的numpy.cos。

参数:

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

返回:

一个包含 x 中每个元素的余弦值的数组，会提升到非精确的 dtype。

返回类型:

数组

参见

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