PyTorch中的pad函数是一个用于调整输入维度的重要函数,它可以在输入维度的所有轴上添加填充值,以满足输入维度的要求。pad函数是torch.nn.functional.pad(),它接受一个输入张量,一个长度为4的元组,元组中的每个值代表在输入张量的每个维度上的填充值,以及一个可选的填充值。
使用方法
pad函数的使用方法如下:
import torch # 定义一个输入张量 input = torch.randn(3, 4, 5) # 定义一个长度为4的元组,元组中的每个值代表在输入张量的每个维度上的填充值 pad = (1, 2, 3, 4) # 定义一个可选的填充值 value = 0 # 调用pad函数 output = torch.nn.functional.pad(input, pad, value)
在上面的代码中,我们定义了一个输入张量input,一个长度为4的元组pad,以及一个可选的填充值value,调用pad函数,它会在输入张量的每个维度上添加pad中指定的填充值,返回一个新的输出张量output。
参数说明
- input:输入张量,形状为[N, C, H, W],其中N表示batch size,C表示通道数,H表示高度,W表示宽度。
- pad:长度为4的元组,元组中的每个值代表在输入张量的每个维度上的填充值,依次为[pad_left, pad_right, pad_top, pad_bottom]。
- value:可选的填充值,默认为0。
- output:输出张量,形状为[N, C, H+pad_top+pad_bottom, W+pad_left+pad_right]。