jax.lax.sub

Contents

jax.lax.sub#

jax.lax.sub(x, y)[source]#

Elementwise subtraction: \(x - y\).

Parameters:
  • x (jax.typing.ArrayLike)

  • y (jax.typing.ArrayLike)

Return type:

Array