jax.numpy.subtract

Contents

jax.numpy.subtract#

jax.numpy.subtract(x, y, /)[source]#

Subtract arguments, element-wise.

LAX-backend implementation of numpy.subtract().

Original docstring below.

Returns:

y – The difference of x1 and x2, element-wise. This is a scalar if both x1 and x2 are scalars.

Return type:

ndarray

Parameters:
  • x (jax.typing.ArrayLike)

  • y (jax.typing.ArrayLike)