Mul
Performs element-wise binary multiplication (with Numpy-style broadcasting support). This operator supports multidirectional (i.e., Numpy-style) broadcasting; for more details please check [the doc](Broadcasting.md). (Opset 14 change): Extend supported types to include uint8, int8, uint16, and iβ¦
Abstract Signature:
Mul(A: Tensor, B: Tensor)
PyTorch
API: torch.mul
Strategy: Direct Mapping
JAX (Core)
API: jnp.multiply
Strategy: Direct Mapping
NumPy
API: np.multiply
Strategy: Infix (*)
Keras
API: keras.ops.multiply
Strategy: Direct Mapping
TensorFlow
API: tf.math.multiply
Strategy: Direct Mapping
Apple MLX
API: mx.multiply
Strategy: Direct Mapping
NVIDIA SASS
API: FMUL
Strategy: Direct Mapping
AMD RDNA
API: v_mul_f32
Strategy: Direct Mapping
Flax NNX
API: optax.tree.mul
Strategy: Direct Mapping