Addcmul

Performs the element-wise multiplication of tensor1 by tensor2, multiplies the result by the scalar value and adds it to input.

Abstract Signature:

Addcmul(input: Tensor, tensor1: Tensor, tensor2: Tensor, value: float = 1.0)

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: torch.addcmul

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Macro 'jnp.add({input}, {value} * jnp.multiply({tensor1}, {tensor2}))'

NumPy

API: —

Strategy: Macro '{input} + {value} * {tensor1} * {tensor2}'

Keras

API: —

Strategy: Macro 'keras.ops.add({input}, keras.ops.multiply({tensor1}, {tensor2}) * {value})'

TensorFlow

API: —

Strategy: Macro 'tf.math.add({input}, tf.math.multiply({tensor1}, {tensor2}) * {value})'

Apple MLX

API: —

Strategy: Macro 'mx.add({input}, mx.multiply({tensor1}, {tensor2}) * {value})'

Flax NNX

API: —

Strategy: Macro 'jnp.add({input}, {value} * jnp.multiply({tensor1}, {tensor2}))'

PaxML / Praxis

API: —

Strategy: Macro 'jnp.add({input}, {value} * jnp.multiply({tensor1}, {tensor2}))'