torch.nn.functional.pad函数介绍

torch.nn.functional.pad 是 PyTorch 中用于对张量进行边界填充的函数，支持多种填充模式，可以在多种场景（如卷积神经网络的输入预处理）中使用。

---

函数签名

torch.nn.functional.pad(
    input: Tensor,
    pad: Tuple[int, ...],
    mode: str = 'constant',
    value: float = 0
) -> Tensor

参数说明

1. input (Tensor):

   • 输入的张量。
   • 可以是任意维度的张量。

2. pad (tuple):

   • 指定每个维度填充的大小。
   • 该元组的长度必须是偶数，表示从最后一个维度开始，依次为每个维度添加填充，按照 (dim_last_before_last, dim_last, dim_second_to_last_before_last, dim_second_to_last, ...)的顺序排列。
   • 每个维度需要两个数字：第一个表示左侧填充，第二个表示右侧填充。
   • 示例： 对 3D 张量 (D, H, W)：
     • pad=(1, 2, 3, 4)：
       • 对 W 维度左填充 1，右填充 2。
       • 对