jax.lax.square

Warning

This page was created from a pull request (#9655).

jax.lax.square¶

jax.lax.square(x)[source]¶

Elementwise square: \(x^2\).

Parameters

x (Any) –

Return type

Any

previous

jax.lax.sqrt

next

jax.lax.squeeze