jax.lax.atan#
- jax.lax.atan(x)[source]#
Elementwise arc tangent: \(\mathrm{atan}(x)\).
This function lowers directly to the
chlo.atan
operation.- Parameters:
x (ArrayLike) – input array. Must have floating-point or complex type.
- Returns:
Array of the same shape and dtype as
x
containing the element-wise arc tangent.- Return type:
See also
jax.lax.tan()
: elementwise tangent.jax.lax.acos()
: elementwise arc cosine.jax.lax.asin()
: elementwise arc sine.jax.lax.atan2()
: elementwise 2-term arc tangent.