jax_subtract

probly.representation.jax_functions.jax_subtract(x1: JaxArrayLike, x2: JaxArrayLike, /) → jax.Array[source]
probly.representation.jax_functions.jax_subtract(x1: JaxOperand, x2: JaxOperand, /) → object

Subtract operands elementwise, mirroring jax.numpy.subtract.

Parameters:

  • x1 – Operand from which to subtract.

  • x2 – Operand to subtract.

Returns:

x1 - x2 elementwise, or the result supplied by a custom override.