PyTorch数学运算详解-优快云博客

本文链接：https://blog.youkuaiyun.com/acm_durante/article/details/122671700

前言：

本文讲解有关Tensor的基础数学运算

一、基础运算

1、相关函数

加法:a + b或torch.add(a,b)
减法:a - b或torch.sub(a,b)
乘法:a * b或torch.mul(a,b)
除法:a / b或torch.div(a,b)

2、代码实例

a = torch.rand(3,4)
b = torch.rand(4)

print(torch.all(torch.eq(a-b,torch.sub(a,b))))
print(torch.all(torch.eq(a+b,torch.add(a,b))))
print(torch.all(torch.eq(a*b,torch.mul(a,b))))
print(torch.all(torch.eq(a/b,torch.div(a,b))))
# 输出是: tensor(True)

二、矩阵运算

1、相关函数

torch.mm(a,b):矩阵乘法，适用于dim=2
torch.matmul(a,b):矩阵乘法，适用范围广泛，并支持Broadcasting机制(传送门)
a @ b:矩阵乘法，与torch.matmul(a,b)用法相同

2、代码实例

dim=2的例子

a = torch.FloatTensor([[3,3],[3,3]])
b = torch.ones(2,2)
c = a * b # element_wise，这里是.*，不是matrix乘法
print(c)
# tensor([[3., 3.],
#        [3., 3.]])

c = torch.mm(a,b) # 只适用于dim=2
print(c)
# tensor([[6., 6.],
#        [6., 6.]])

c = torch.matmul(a,b)
print(c)
# tensor([[6., 6.],
#        [6., 6.]])

c = a @ b
print(c)
# tensor([[6., 6.],
#        [6., 6.]])

在神经网络中完成降维过程

# 神经网络中完成将维度的过程 [4,784] => [4,512]
a = torch.rand(4,784)
x = torch.rand(4,784)
w = torch.rand(512,784)
ans = x @ w.t() #.t()只适合2d
print(ans.shape)

高dim的matmul实例

a = torch.rand(4,3,28,64)
b = torch.rand(4,3,64,32)
c = torch.matmul(a,b) # 前两维度不变，后两维度做matmul 
print(c.shape)
# out : torch.Size([4, 3, 28, 32])

b = torch.rand(4,1,64,32) # 后两维度做matmul，前两维度做Broadcasting机制b变换成[4,3]
c = torch.matmul(a,b) 
print(c.shape)
# out: torch.Size([4, 3, 28, 32])

三、power函数

1、相关函数

a.pow(x):完成 $a^{x}$ 运算
a ** x:完成 $a^{x}$ 运算
a.sqrt():完成 $a^{0.5}$ 运算
`a.rsqrt()``:完成平方根导数运算

2、代码实例

a = torch.full([2,2],3)
#  tensor([[3., 3.],
#         [3., 3.]])

b = a.pow(2)
print(b)
# tensor([[9., 9.],
#         [9., 9.]])

b = a ** 2
print(b)
# tensor([[9., 9.],
#         [9., 9.]])

c = b.sqrt() # 平方根
print(c)
# tensor([[3., 3.],
#        [3., 3.]])

c = b.rsqrt() # 平方根导数
print(c)
# tensor([[0.3333, 0.3333],
#         [0.3333, 0.3333]])

c = b ** (0.5) # sqrt
print(c)
# tensor([[3., 3.],
#        [3., 3.]])

四、exp函数与log函数

1、相关函数

torch.exp(tensor):对于每一个数，都取 $e^x$
torch.log(tensor):对于每一个数，都取 $log_ex$ ，该函数默认e为底，若想以其他为底，可以使用torch.log2等，或者使用换底公式

2、代码实例

a = torch.exp(torch.ones(2,2)) # e^1
# tensor([[2.7183, 2.7183],
#        [2.7183, 2.7183]])

b = torch.log(a)
print(b)
# tensor([[1., 1.],
#         [1., 1.]])

五、近似函数

1、相关函数

a.floor():向下取整
a.ceil():向上取整
a.trunc():取整数部分
a.frac():取小数部分
a.round():四舍五入

2、代码实例

a = torch.tensor(3.1415)
print(a.floor()) # 向下取整 tensor(3.)
print(a.ceil()) # 向上取整 tensor(4.)
print(a.trunc()) # 取整数部分 tensor(3.)
print(a.frac()) # 取小数部分 tensor(0.1415)
# --------------------------------------------
a = torch.tensor(3.499)
print(a.round()) # tensor(3.)
a = torch.tensor(3.5)
print(a.round()) # tensor(4.)

六、clamp函数

1、相关函数

a.max():找到最大的数
a.median():找到中位数
a.clamp(min,max):将范围设置为[min,max]之间，不在其范围内的，置成最大值或最小值。若只输入一个数，则代表输入的是min，max取正无穷

2、代码实例

grad = torch.rand(2,3)*15
print(grad)
# tensor([[ 1.3345, 10.6758, 12.4728],
#        [ 7.3111,  0.3316,  8.4921]])

print(grad.max())
# tensor(12.4728)

print(grad.median())
# tensor(7.3111)

print(grad.clamp(10)) # 小于10置为10
# tensor([[10.0000, 10.6758, 12.4728],
#        [10.0000, 10.0000, 10.0000]])

print(grad.clamp(0,10)) # 限制在[0,10]
# tensor([[ 1.3345, 10.0000, 10.0000],
#        [ 7.3111,  0.3316,  8.4921]])