jax.numpy.cos#

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

Compute a trigonometric cosine of each element of input.

JAX implementation ofnumpy.cos.

Parameters:

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

Returns:

An array containing the cosine of each element inx, promotes to inexactdtype.

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 ofinput.

• jax.numpy.arccos() andjax.numpy.acos(): Computes the inverse oftrigonometric cosine of each element of input.

Examples

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