PyTorch torch.mul 函数
Pytorch torch 参考手册torch.mul 是 PyTorch 中用于执行逐元素乘法的函数。它将两个张量或一个张量和一个标量相乘。
这是深度学习中最基础的数学运算之一。
函数定义
torch.mul(input, other, out=None)
参数:
input(Tensor): 第一个输入张量。other(Tensor 或 float): 第二个输入张量或标量。out(Tensor, 可选): 输出张量。
返回值:
torch.Tensor: 返回相乘后的张量。
使用示例
示例 1: 张量逐元素相乘
实例
import torch
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])
c = torch.mul(a, b)
print(c)
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])
c = torch.mul(a, b)
print(c)
输出结果为：
tensor([ 4, 10, 18])
示例 2: 乘以标量
实例
import torch
a = torch.tensor([1, 2, 3])
# 乘以 2
b = torch.mul(a, 2)
print(b)
# 也可以使用 * 运算符
c = a * 2
print(c)
a = torch.tensor([1, 2, 3])
# 乘以 2
b = torch.mul(a, 2)
print(b)
# 也可以使用 * 运算符
c = a * 2
print(c)
输出结果为：
tensor([2, 4, 6]) tensor([2, 4, 6])