Pytorch常用函数用法归纳:Tensor张量之间的计算

作者：Cpp五条 | 2024-06-06 09:24:27

踩

1.torch.add()

(1)函数原型:

torch.add(input, other, alpha, out)

(2)参数说明:

参数名称	参数类型	参数说明
input	torch.Tensor	表示参与运算的第一个输入Tensor张量
other	torch.Tensor或者Number	表示参与运算的第二个输入Tensor张量或标量
alpha	Number, optional	一个可选的缩放因子，默认为1，other张量将会乘以这个因子后再进行加法运算
out	torch.Tensor, optional	用于存储输出结果的张量。如果提供了该参数,结果将存储在这个张量中;默认值为None

(3)函数功能:

对输入张量input逐元素加上标量值/张量other,并返回一个新的张量tensor;运算过程中需要满足广播机制

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/Cpp五条/article/detail/680412