torch.mul — PyTorch 2.5 documentation (original) (raw)

torch.mul(input, other, *, out=None) → Tensor¶

Multiplies input by other.

outi=inputi×otheri\text{out}_i = \text{input}_i \times \text{other}_i

Supports broadcasting to a common shape,type promotion, and integer, float, and complex inputs.

Parameters

• input (Tensor) – the input tensor.
• other (Tensor or Number) –

Keyword Arguments

out (Tensor, optional) – the output tensor.

Examples:

a = torch.randn(3) a tensor([ 0.2015, -0.4255, 2.6087]) torch.mul(a, 100) tensor([ 20.1494, -42.5491, 260.8663])

b = torch.randn(4, 1) b tensor([[ 1.1207], [-0.3137], [ 0.0700], [ 0.8378]]) c = torch.randn(1, 4) c tensor([[ 0.5146, 0.1216, -0.5244, 2.2382]]) torch.mul(b, c) tensor([[ 0.5767, 0.1363, -0.5877, 2.5083], [-0.1614, -0.0382, 0.1645, -0.7021], [ 0.0360, 0.0085, -0.0367, 0.1567], [ 0.4312, 0.1019, -0.4394, 1.8753]])