torch.nn.functional.pad使用详解

torch.nn.functional.pad 是 PyTorch 中用于对张量进行填充的函数。填充（Padding）指在张量的边缘添加额外元素的操作，常见于图像处理等场景（例如为图像添加边框）。

函数定义
torch.nn.functional.pad(input, pad, mode='constant', value=None) → Tensor

参数说明

• input (Tensor): 需要填充的N维张量
• pad (tuple): 指定各维度填充数量的元组。长度必须为偶数。
• mode (str): 填充模式，可选：
  • 'constant': 用固定值填充（默认0）
  • 'reflect': 边缘镜像反射填充
  • 'replicate': 边缘复制填充
  • 'circular': 循环重复填充
• value (Optional[float]): 'constant'模式下的填充值，默认为0

返回值
Tensor: 填充后的张量

示例

示例1：最后一维填充