jax.lax.abs

jax.lax.abs(x)[source]

Elementwise absolute value: \(|x|\).

Parameters

x (Any) –

Return type

Any