Python PyTorch addcdiv用法及代码示例

本文简要介绍python语言中 torch.addcdiv 的用法。

用法:

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

参数：

• input(Tensor) -要添加的张量

• tensor1(Tensor) -分子张量

• tensor2(Tensor) -分母张量

关键字参数：

• value(数字,可选的) -\text{tensor1} / \text{tensor2} 的乘数

• out(Tensor,可选的) -输出张量。

执行 tensor1 除以 tensor2 的元素除法，将结果乘以标量 value 并将其添加到 input 。

警告

不再支持使用 addcdiv 进行整数除法，在未来的版本中，addcdiv 将执行 tensor1 和 tensor2 的真正除法。历史 addcdiv 行为可以实现为 (input + value * torch.trunc(tensor1 /tensor2)).to(input.dtype) 用于整数输入和 (input + value * tensor1 /tensor2) 用于浮点输入。未来的 addcdiv 行为只是后一种实现：(input + value * tensor1 /tensor2)，适用于所有 dtype。

\text{out}_i = \text{input}_i + \text{value} \times \frac{\text{tensor1}_i}{\text{tensor2}_i}

input 、 tensor1 和 tensor2 的形状必须是可广播的。

对于 FloatTensor 或 DoubleTensor 类型的输入，value 必须为实数，否则为整数。

例子：

>>> t = torch.randn(1, 3)
>>> t1 = torch.randn(3, 1)
>>> t2 = torch.randn(1, 3)
>>> torch.addcdiv(t, t1, t2, value=0.1)
tensor([[-0.2312, -3.6496,  0.1312],
        [-1.0428,  3.4292, -0.1030],
        [-0.5369, -0.9829,  0.0430]])