本文主要是介绍Pytorch常用函数用法归纳:Tensor张量之间的计算,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
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;运算过程中需要满足广播机制
这篇关于Pytorch常用函数用法归纳:Tensor张量之间的计算的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!