加减乘除
矩阵相乘
torch.mm 只适用于2d,不推荐
torch.matmul 推荐使用
@ 运算符重载
次方运算
a = torch.full([2,2],3)
aa = a.pow(2)
操作运算如下:
a.pow(n)
表示an
a**n
表示an
aa.sqrt()
表示aa 1/2
aa.rsqrt()
表示aa1/3
注意这里没有torch.
幂/对数
a = torch.exp(torch.one(2,2)) #指数运算
b = torch.log(a) #对数运算,默认底数为e
c = torch.log2(a) #对数运算,改变底数
近似值
向下取整,floor()
向上取整,ceil()
取整