F.pad函数、torch.chunk函数、torch.roll函数、torch.narrow函数

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, 可选): 当mode='constant'时，用于填充的值。默认是0。

下面是一个二维张量（模拟灰度图像）的填充示例