jax.numpy.subtract#
- jax.numpy.subtract(x1, x2, /)[source]#
Subtract arguments, element-wise.
LAX-backend implementation of
numpy.subtract().Original docstring below.
- Parameters:
x1 (array_like) – The arrays to be subtracted from each other. If
x1.shape != x2.shape, they must be broadcastable to a common shape (which becomes the shape of the output).x2 (array_like) – The arrays to be subtracted from each other. If
x1.shape != x2.shape, they must be broadcastable to a common shape (which becomes the shape of the output).
- Returns:
y – The difference of x1 and x2, element-wise. This is a scalar if both x1 and x2 are scalars.
- Return type:
ndarray