Dist

Returns the p-norm of (input - other).

Abstract Signature:

Dist(input: Tensor, other: Tensor, p: float = 2.0)

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: torch.dist

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.numpy.linalg.norm

Strategy: Macro 'jax.numpy.linalg.norm({input} - {other}, ord={p})'

NumPy

API: numpy.linalg.norm

Strategy: Macro 'numpy.linalg.norm({input} - {other}, ord={p})'

Keras

API: keras.ops.norm

Strategy: Macro 'keras.ops.norm({input} - {other}, ord={p})'

TensorFlow

API: tf.norm

Strategy: Macro 'tf.norm({input} - {other}, ord={p})'

Apple MLX

API: mlx.core.linalg.norm

Strategy: Macro 'mlx.core.linalg.norm({input} - {other}, ord={p})'

Flax NNX

API: jax.numpy.linalg.norm

Strategy: Macro 'jax.numpy.linalg.norm({input} - {other}, ord={p})'

PaxML / Praxis

API: jax.numpy.linalg.norm

Strategy: Macro 'jax.numpy.linalg.norm({input} - {other}, ord={p})'