从零开始学习Pytorch之张量的数学运算

一、加减乘除

torch.add(input, alpha=1, other, out=None)

功能：逐元素计算input+alpha×other
·input：第一个张量
·alpha：乘项因子
·other：第二个张量

t1 = torch.randn((3,3))#3×3的标准正态分布矩阵
t2 = torch.ones_like(t1)
t_add = torch.add(t1, 10, t2)
print('t1:\n{}\nt2:\n{}\nt_add:\n{}'.format(t1,t2,t_add))

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

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

torch.sub()
torch.div()
torch.mul()

二、对数，指数，幂函数

torch.log(input, out=None)
torch.log10(input, out=None)
torch.log2(input, out=None)
torch.exp(input, out=None)
torch.pow()

三、三角函数

torch.abs(input, out=None)
torch.acos(input, out=None)
torch.cosh(input, out=None)
torch.cos(input, out=None)
torch.asin(input, out=None)
torch.atan(input, out=None)
torch.atan2(input, other, out=None)

PS.这篇博客只列举出了函数，并对常用到的进行举例，实际运用可以现用现查。