【Python】nn.nn.CircularPad1、2、3d函数和nn.ConstantPad1、2、3d函数详解和示例

前言

在深度学习中，尤其是在处理图像、音频或其他多维数据时，数据填充（Padding）是一个常见的操作。填充不仅可以保持数据的空间维度，还能在卷积操作中避免信息丢失。PyTorch提供了多种填充方式，其中nn.CircularPadXd和nn.ConstantPadXd是两种常用的填充方法。本文将详细介绍nn.CircularPad1d、nn.CircularPad2d、nn.CircularPad3d以及nn.ConstantPad1d、nn.ConstantPad2d、nn.ConstantPad3d的函数原型、用法示例，并总结它们的特点。

函数原型

nn.CircularPadXd

nn.CircularPadXd（其中X可以是1、2或3）类用于使用循环填充（circular padding）方式填充输入张量（tensor）的边界。循环填充意味着在填充时使用输入张量边界的值来填充另一端。

nn.CircularPad1d(padding): 一维数据的循环填充。
nn.CircularPad2d(padding): 二维数据的循环填充，常用于图像。
nn.CircularPad3d(padding): 三维数据的循环填充，适用于视频帧或体积数据。
padding参数可以是一个整数或元组（tuple），表示在每个维度上的填充大小。

nn.ConstantPadXd

nn.ConstantPadXd（其中X可以是1、2或3）类用于使用常数填充（constant padding）方式填充输入张量的边界。

nn.ConstantPad1d(padding, value): 一维数据的常数填充。
nn.ConstantPad2d(padding, value): 二维数据的常数填充，常用于图像。
nn.ConstantPad3d(padding, value): 三维数据的常数填充，适用于视频帧或体积数据。
padding参数同上，value参数指定了填充的常数值。

示例

nn.CircularPad1、2、3d 示例

import torch
import torch.nn as nn

# 示例：nn.CircularPad2d
input = torch.arange(9, dtype=torch.float).reshape(1, 1, 3, 3)
print("Input:")
print(input)

pad = nn.CircularPad2d(1)
output = pad(input)
print("Output with CircularPad2d:")
print(output)

# 使用不同的填充大小
pad = nn.CircularPad2d((1, 2, 0, 1))
output = pad(input)
print("Output with different padding:")
print(output)

nn.ConstantPad1、2、3d 示例

import torch
import torch.nn as nn
# 示例：nn.ConstantPad2d
input = torch.arange(9, dtype=torch.float).reshape(1, 1, 3, 3)
print("Input:")
print(input)

pad = nn.ConstantPad2d((1, 2, 0, 1), value=10.0)
output = pad(input)
print("Output with ConstantPad2d:")
print(output)
print("***********************************")
# 三维数据示例
input_3d = torch.randn(1, 1, 2, 3, 4)
pad_3d = nn.ConstantPad3d((1, 2, 0, 1, 2, 3), value=5.0)
output_3d = pad_3d(input_3d)
print("Output with ConstantPad3d:")
print(output_3d)

小结

nn.CircularPadXd：使用循环填充方式，通过在张量边界值之间循环来填充数据。这对于需要保持数据连续性的场景特别有用。
nn.ConstantPadXd：使用指定的常数值来填充张量的边界。这种填充方式简单直接，适用于不需要特殊边界处理的场景。

无论是循环填充还是常数填充，都是深度学习中常用的数据预处理技术，它们能够帮助我们更好地控制数据的维度和边界条件，从而优化模型的训练效果。