jax.numpy.cos — JAX documentation (original) (raw)

jax.numpy.cos#

jax.numpy.cos(x, /)[source]#

Compute a trigonometric cosine of each element of input.

JAX implementation of numpy.cos.

Parameters:

x (ArrayLike) – scalar or array. Angle in radians.

Returns:

An array containing the cosine of each element in x, promotes to inexact dtype.

Return type:

Array

See also

• jax.numpy.sin(): Computes a trigonometric sine of each element of input.
• jax.numpy.tan(): Computes a trigonometric tangent of each element of input.
• jax.numpy.arccos() and jax.numpy.acos(): Computes the inverse of trigonometric cosine of each element of input.

Examples

pi = jnp.pi x = jnp.array([pi/4, pi/2, 3pi/4, 5pi/6]) with jnp.printoptions(precision=3, suppress=True): ... print(jnp.cos(x)) [ 0.707 -0. -0.707 -0.866]