评价此页

torch.addcdiv#

torch.addcdiv(input, tensor1, tensor2, *, value=1, out=None) 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。

outi=inputi+value×tensor1itensor2i\text{out}_i = \text{input}_i + \text{value} \times \frac{\text{tensor1}_i}{\text{tensor2}_i}

inputtensor1tensor2 的形状必须可广播

对于 FloatTensorDoubleTensor 类型的输入,value 必须是实数,否则为整数。

参数
  • input (Tensor) – 要添加的张量

  • tensor1 (Tensor) – 分子张量

  • tensor2 (Tensor) – 分母张量

关键字参数
  • value (Number, optional) – tensor1/tensor2\text{tensor1} / \text{tensor2} 的乘数

  • out (Tensor, optional) – 输出张量。

示例

>>> 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]])