欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2022-07-14 11:02:59
...

一、加减乘除

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

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

torch.addcmul(input, value=1, tensor1, tensor2, out=None)
从零开始学习Pytorch之张量的数学运算
torch.addcdiv(input, value=1, tensor1, tensor2, out=None)
从零开始学习Pytorch之张量的数学运算
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.这篇博客只列举出了函数,并对常用到的进行举例,实际运用可以现用现查。