SquaredDifference

Returns (x - y)(x - y) element-wise.

Abstract Signature:

SquaredDifference(x: Tensor, y: Tensor)

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: —

Strategy: Macro 'torch.square({x} - {y})'

JAX (Core)

API: —

Strategy: Macro 'jnp.square({x} - {y})'

NumPy

API: —

Strategy: Macro 'numpy.square({x} - {y})'

Keras

API: —

Strategy: Macro 'keras.ops.square({x} - {y})'

TensorFlow

API: tf.math.squared_difference

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: —

Strategy: Macro 'mlx.core.square({x} - {y})'

Flax NNX

API: —

Strategy: Macro 'jnp.square({x} - {y})'

PaxML / Praxis

API: —

Strategy: Macro 'jnp.square({x} - {y})'