1、F.pad函数
torch.nn.functional.pad(input, pad, mode='constant', value=0)
函数变量说明:
- input (
Tensor
): 需要填充的张量。 - pad (
tuple[int, int]
或tuple[tuple[int, int], ...]
): 一个整数元组,指定在每个维度上要填充的零填充的数量。对于二维张量(如图像),这个元组通常是(pad_left, pad_right, pad_top, pad_bottom)
。对于更高维度的张量,可以嵌套元组来指定每个维度的填充。 - mode (
str
, 可选): 填充的模式。默认是'constant'
,表示用常数填充。其他选项可能包括'reflect'
(反射填充,即镜像边缘的值)、'replicate'
(复制填充,即复制边缘的值)等。 - value (
float<