jax.lax.cos

jax.lax.cos(x, accuracy=None)

Elementwise cosine: \(\mathrm{cos}(x)\).

For floating-point inputs, this function lowers directly to the stablehlo.cosine operation. For complex inputs, it lowers to a sequence of HLO operations implementing the complex cosine.

Parameters:

• x (ArrayLike) – input array. Must have floating-point or complex type.

• accuracy – Optional lax.Tolerance or lax.AccuracyMode object that selects the implementation of the op based on the requested accuracy. If the implementation cannot satisfy the requested tolerance, the compiler will return an error. If mode is specified and there are no multiple implementations available, the default implementation will be used.

Returns:

Array of the same shape and dtype as x containing the element-wise cosine.

Return type:

Array

See also

• jax.lax.sin(): elementwise sine.

• jax.lax.tan(): elementwise tangent.

• jax.lax.acos(): elementwise arc cosine.