jax.lax.square#
- jax.lax.square(x)[source]#
Elementwise square: \(x^2\).
- Numerical Precision:
For accuracy bounds, see Numerical accuracy.
- Parameters:
x (ArrayLike)
- Return type:
Elementwise square: \(x^2\).
For accuracy bounds, see Numerical accuracy.
x (ArrayLike)