ScalarMul

Multiplies a scalar times a Tensor.

Abstract Signature:

ScalarMul(scalar: float, x: Tensor)

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: torch.mul

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Macro '{x} * {scalar}'

NumPy

API: —

Strategy: Macro '{x} * {scalar}'

Keras

API: —

Strategy: Macro '{x} * {scalar}'

TensorFlow

API: tf.math.scalar_mul

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: —

Strategy: Macro '{x} * {scalar}'

Flax NNX

API: —

Strategy: Macro '{x} * {scalar}'

PaxML / Praxis

API: —

Strategy: Macro '{x} * {scalar}'