def test_groupnorm(self):
    self._check_one_layer(nn.GroupNorm(4, 16), torch.randn(64, 16, 10))
    self._check_one_layer(nn.GroupNorm(4, 16), torch.randn(64, 16, 10, 9))
    self._check_one_layer(nn.GroupNorm(4, 16), torch.randn(64, 16, 10, 9, 8))
def test_groupnorm(self):
    self._check_one_layer(nn.GroupNorm(4, 16), torch.randn(64, 16, 10))
    self._check_one_layer(nn.GroupNorm(4, 16), torch.randn(64, 16, 10, 9))
    self._check_one_layer(nn.GroupNorm(4, 16), torch.randn(64, 16, 10, 9, 8))