UpsamplingNearest2d¶

class dragon.vm.torch.nn.UpsamplingNearest2d(
  size=None,
  scale_factor=None
)[source]¶

Upsample input via nearest interpolating.

Set size or scale_factor to determine the output size:

x = torch.ones((1, 2, 3, 4))
y = torch.nn.UpsamplingNearest2d(size=6)(x)  # Size: (1, 2, 6, 6)
z = torch.nn.UpsamplingNearest2d(scale_factor=2)(x)  # Size: (1, 2, 6, 8)

See also

torch.nn.functional.interpolate(…)