Sub

Subtracts other from input.

Abstract Signature:

Sub(input: Tensor, other: Tensor, alpha: float = 1.0)

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLXNVIDIA SASSAMD RDNAFlax NNX

PyTorch

API: torch.sub

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: optax.tree.sub

Strategy: Macro 'jax.numpy.subtract({input}, {other} * {alpha})'

NumPy

API: numpy.f2py.symbolic.ArithOp.SUB

Strategy: Macro 'numpy.subtract({input}, {other} * {alpha})'

Keras

API: keras.ops.subtract

Strategy: Macro 'keras.ops.subtract({input}, {other} * {alpha})'

TensorFlow

API: tf.math.subtract

Strategy: Macro 'tf.math.subtract({input}, {other} * {alpha})'

Apple MLX

API: mx.subtract

Strategy: Macro 'mlx.core.subtract({input}, {other} * {alpha})'

NVIDIA SASS

API: FADD

Strategy: Direct Mapping

AMD RDNA

API: v_sub_f32

Strategy: Direct Mapping

Official Docs ↗

Flax NNX

API: optax.tree.sub

Strategy: Direct Mapping

Official Docs ↗