最近邻上采样2D¶
- class torch.nn.UpsamplingNearest2d(size=None, scale_factor=None)[source]¶
将 2D 最近邻上采样应用于由多个输入通道组成的输入信号。
要指定比例,它将
size
或scale_factor
作为其构造函数参数。当给出
size
时,它是图像的输出大小 (h, w)。- 参数
警告
此类已弃用,建议使用
interpolate()
。- 形状
输入:
输出: 其中
示例
>>> input = torch.arange(1, 5, dtype=torch.float32).view(1, 1, 2, 2) >>> input tensor([[[[1., 2.], [3., 4.]]]]) >>> m = nn.UpsamplingNearest2d(scale_factor=2) >>> m(input) tensor([[[[1., 1., 2., 2.], [1., 1., 2., 2.], [3., 3., 4., 4.], [3., 3., 4., 4.]]]])