一、介绍
在深度学习中，我们常常需要对输入的张量进行填充操作，以便处理不同大小的输入数据。在PyTorch中，我们可以使用torch.nn.functional.pad函数来实现张量的填充功能。本文将介绍如何使用torch.nn.functional.pad函数对张量进行填充操作。
二、使用torch.nn.functional.pad进行填充
使用torch.nn.functional.pad函数前，我们需要先了解函数的参数。torch.nn.functional.pad函数的参数如下：
torch.nn.functional.pad(input, pad, mode='constant', value=0)
其中，参数input是要进行填充的张量；参数pad是填充的大小，它可以是一个整数，也可以是一个元组；参数mode表示填充的方式，默认是’constant’，表示用常数填充；参数value表示填充的数值，默认是0。
下面是一个简单的例子，我们对一个2 x 2的张量进行大小为1的填充：
import torch x = torch.ones(2, 2) print(x) x = torch.nn.functional.pad(x, (1, 1, 1, 1)) print(x)
运行结果如下所示：
tensor([[1., 1.],
[1., 1.]])
tensor([[0., 0., 0., 0., 0.],
[0., 1., 1., 1., 0.],
[0., 1., 1., 1., 0.],
[0., 1., 1., 1., 0.],
[0., 0., 0., 0., 0.]])
可以看出，填充后的张量大小为4 x 4，周围都填充了1列（或1行）的0。
三、填充的方式和数值
torch.nn.functional.pad函数支持不同的填充方式和不同的填充数值。
1、填充方式
除了常数填充(‘constant’)外，torch.nn.functional.pad函数还支持以下填充方式：
- ‘reflect’：以边缘为轴，对称填充。
- ‘replicate’：以边缘为轴，复制填充。
- ‘circular’：循环填充。
例如，如果我们使用’reflect’填充上面的例子，代码如下所示：
import torch x = torch.ones(2, 2) print(x) x = torch.nn.functional.pad(x, (1, 1, 1, 1), mode='reflect') print(x)
运行结果如下所示：
tensor([[1., 1.],
[1., 1.]])
tensor([[1., 1., 1., 1., 1.],
[1., 1., 1., 1., 1.],
[1., 1., 1., 1., 1.],
[1., 1., 1., 1., 1.],
[1., 1., 1., 1., 1.]])
可以看出，使用’reflect’填充后，边缘处的值被进行了对称填充。
2、填充数值
我们还可以通过value参数来指定填充的数值。
以填充值为1为例，代码如下所示：
import torch x = torch.ones(2, 2) print(x) x = torch.nn.functional.pad(x, (1, 1, 1, 1), value=1) print(x)
运行结果如下所示：
tensor([[1., 1.],
[1., 1.]])
tensor([[1., 1., 1., 1., 1.],
[1., 1., 1., 1., 1.],
[1., 1., 1., 1., 1.],
[1., 1., 1., 1., 1.],
[1., 1., 1., 1., 1.]])
可以看出，填充后的张量中，原张量部分的值为1，填充部分的值也为1。
四、指定不同的填充大小
我们还可以在不同的维度上使用不同的填充大小。
例如，如果我们在第1维度和第2维度上分别使用不同的填充大小，代码如下所示：
import torch x = torch.ones(2, 2) print(x) x = torch.nn.functional.pad(x, (1, 2, 1, 0), value=1) print(x)
运行结果如下所示：
tensor([[1., 1.],
[1., 1.]])
tensor([[1., 1., 1., 1., 1., 1.],
[1., 1., 1., 1., 1., 1.],
[1., 1., 1., 1., 1., 0.],
[1., 1., 1., 1., 1., 0.]])
可以看出，第1维度上的填充大小为(1, 2)，第2维度上的填充大小为(1, 0)。
五、结论
torch.nn.functional.pad函数是PyTorch中用于进行张量填充的函数，能够支持不同的填充方式和不同的填充数值。在使用时，需要注意填充的大小可以是一个整数，也可以是一个元组；填充的方式可以是’constant’、’reflect’、’replicate’或者’circular’；填充的数值可以通过value参数进行指定。同时，我们还可以在不同的维度上使用不同的填充大小。
原创文章，作者：小蓝，如若转载，请注明出处：https://www.506064.com/n/199334.html
微信扫一扫 
支付宝扫一扫 