PyTorch torch.addcmul 函数
Pytorch torch 参考手册torch.addcmul 是 PyTorch 中用于执行 input + value * (tensor1 * tensor2) 逐元素运算的函数。
函数定义
torch.addcmul(input, tensor1, tensor2, value=1, out=None)
参数说明
input: 输入张量tensor1: 第一个乘数张量tensor2: 第二个乘数张量value: 乘数，默认为1out: 输出张量（可选）
使用示例
实例
import torch
# 创建张量
input = torch.tensor([1.0, 2.0, 3.0])
tensor1 = torch.tensor([2.0, 3.0, 4.0])
tensor2 = torch.tensor([3.0, 4.0, 5.0])
# 执行 input + value * (tensor1 * tensor2)
result = torch.addcmul(input, tensor1, tensor2, value=1)
print(result)
# 创建张量
input = torch.tensor([1.0, 2.0, 3.0])
tensor1 = torch.tensor([2.0, 3.0, 4.0])
tensor2 = torch.tensor([3.0, 4.0, 5.0])
# 执行 input + value * (tensor1 * tensor2)
result = torch.addcmul(input, tensor1, tensor2, value=1)
print(result)
输出结果为：
tensor([ 7., 14., 23.])