SumΒΆ

Returns the sum of all elements in the input tensor.

Abstract Signature:

Sum(input: Tensor, dim: Optional[Union[int, List[int]]], keepdim: bool = False, dtype: Optional[Any])

PyTorch

API: torch.sum
Strategy: Direct Mapping

JAX (Core)

API: jax.numpy.sum
Strategy: Direct Mapping

NumPy

API: numpy.sum
Strategy: Direct Mapping

Keras

API: keras.ops.sum
Strategy: Direct Mapping

TensorFlow

API: tf.math.reduce_sum
Strategy: Direct Mapping

Apple MLX

API: mlx.core.sum
Strategy: Direct Mapping

Flax NNX

API: jax.numpy.sum
Strategy: Direct Mapping

PaxML / Praxis

API: jax.numpy.sum
Strategy: Direct Mapping