现在位置: 首页 > PyTorch 教程 > 正文

PyTorch torch.addcdiv 函数


Pytorch torch 参考手册 Pytorch torch 参考手册

torch.addcdiv 是 PyTorch 中用于执行 input + value * (tensor1 / tensor2) 逐元素运算的函数。

函数定义

torch.addcdiv(input, tensor1, tensor2, value=1, out=None)

参数说明

  • input: 输入张量
  • tensor1: 分子张量
  • tensor2: 分母张量(不能为零)
  • value: 乘数,默认为1
  • out: 输出张量(可选)

使用示例

实例

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)

输出结果为:

tensor([3., 6., 9.])

Pytorch torch 参考手册 Pytorch torch 参考手册