pytorch——张量的数学运算

这里的乘法表示element-wise矩阵中对应位置元素相乘

a=torch.randn(3,4)
b=torch.randn(4)
print(a)
print(b)
print(a+b)
print(torch.add(a,b))
print(a-b)
print(torch.sub(a,b))
print(a*b)
print(torch.mul(a,b))
print(a/b)
print(torch.div(a,b))

matmul表示矩阵乘法，可以对任意维的张量操作，重载的@运算符表示matmul
mm只能针对二维矩阵相乘

a=torch.rand(2,2)
b=torch.rand(2,2)
print(a)
print(b)
print(torch.matmul(a,b))
print(a@b)

matmul如果对高维矩阵操作默认只将最后两维进行相乘，前面的不变

a=torch.rand(2,2)
print(a)
print(a.pow(2)) # 二次方
print(a**2) # 二次方
print(a.sqrt()) # 开根号
print(a.rsqrt()) # 开根号的导数
print(a**(0.5)) # 开根号

a=torch.rand(2,2)
print(a)
print(torch.exp(a)) # e作为底数
print(torch.log(a)) # 默认以e作为底数，想用2做底数用log2

a=torch.tensor(3.14)
print(a.floor()) # 下取整
print(a.ceil()) # 上取整
print(a.trunc()) # 取整数
print(a.frac()) # 取小数
print(a.round()) # 四舍五入

max取最大值
median取最大值
clamp(10)小于10的都变成10
clamp(0,10)不在区间[0,10]之内的都变成10

a=torch.rand(3,4)*15
print(a.max())
print(a.median())
print(a)
print(a.clamp(10))
print(a.clamp(0,10))