PyTorch torch.addcdiv 函数
torch.addcdiv 是 PyTorch 中用于执行 input + value * (tensor1 / tensor2) 逐元素运算的函数。
函数定义
torch.addcdiv(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([4.0, 8.0, 12.0])
tensor2 = torch.tensor([2.0, 2.0, 2.0])
# 执行 input + value * (tensor1 / tensor2)
result = torch.addcdiv(input, tensor1, tensor2, value=1)
print(result)
# 创建张量
input = torch.tensor([1.0, 2.0, 3.0])
tensor1 = torch.tensor([4.0, 8.0, 12.0])
tensor2 = torch.tensor([2.0, 2.0, 2.0])
# 执行 input + value * (tensor1 / tensor2)
result = torch.addcdiv(input, tensor1, tensor2, value=1)
print(result)
输出结果为:
tensor([3., 6., 9.])

Pytorch torch 参考手册