pytorch中的nn.ZeroPad2d()零填充函数实例详解

以下是关于“PyTorch 中的 nn.ZeroPad2d() 零填充函数实例详解”的完整攻略，其中包含两个示例说明。

示例1：使用 nn.ZeroPad2d() 进行零填充

步骤1：导入 PyTorch 库

import torch
import torch.nn as nn

步骤2：创建输入数据

x = torch.randn(1, 3, 4, 4)

在本示例中，我们创建了一个大小为 1x3x4x4 的张量作为输入数据。

步骤3：使用 nn.ZeroPad2d() 进行零填充

pad = nn.ZeroPad2d(2)
x_pad = pad(x)

在本示例中，我们使用 nn.ZeroPad2d() 函数对输入数据进行了 2 个单位的零填充。

示例2：使用 nn.ZeroPad2d() 进行不对称零填充

步骤1：导入 PyTorch 库

import torch
import torch.nn as nn

步骤2：创建输入数据

x = torch.randn(1, 3, 4, 4)

在本示例中，我们创建了一个大小为 1x3x4x4 的张量作为输入数据。

步骤3：使用 nn.ZeroPad2d() 进行不对称零填充

pad = nn.ZeroPad2d((1, 2, 1, 3))
x_pad = pad(x)

在本示例中，我们使用 nn.ZeroPad2d() 函数对输入数据进行了不对称的零填充，其中左侧填充了 1 个单位，右侧填充了 2 个单位，上方填充了 1 个单位，下方填充了 3 个单位。

通过以上步骤，我们可以使用 nn.ZeroPad2d() 函数对输入数据进行零填充，并成功地实现了两个示例。