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

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

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

import torch
import torch.nn as nn

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

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

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

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

import torch
import torch.nn as nn

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

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

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

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

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

展开阅读全文

上一篇：HTML实现圆角边框样式的多种方法及示例代码下一篇：C++ unordered_map emplace()和emplace_hint()方法