Addmv¶
In-place version of Addmv.
Abstract Signature:
Addmv_(input: Tensor, mat: Tensor, vec: Tensor, beta: float = 1.0, alpha: float = 1.0)
JAX (Core)
API:
—Strategy: Macro 'jnp.add({beta} * {input}, {alpha} * jnp.matmul({mat}, {vec}))'
NumPy
API:
—Strategy: Macro '{beta} * {input} + {alpha} * np.matmul({mat}, {vec})'
Keras
API:
—Strategy: Macro 'keras.ops.add({beta} * {input}, {alpha} * keras.ops.matmul({mat}, {vec}))'
TensorFlow
API:
—Strategy: Macro 'tf.math.add({beta} * {input}, {alpha} * tf.linalg.matvec({mat}, {vec}))'
Apple MLX
API:
—Strategy: Macro 'mx.add({beta} * {input}, {alpha} * mx.matmul({mat}, {vec}))'
Flax NNX
API:
—Strategy: Macro 'jnp.add({beta} * {input}, {alpha} * jnp.matmul({mat}, {vec}))'
PaxML / Praxis
API:
—Strategy: Macro 'jnp.add({beta} * {input}, {alpha} * jnp.matmul({mat}, {vec}))'