While InstanceNorm2d should work, I’m not sure if the usage would make sense at all, since you’ll get an all zero output tensor if you just have a single pixel.
norm = nn.InstanceNorm2d(512)
norm(torch.randn(1, 512, 1, 1))