def test_BatchNorm2d(self):
    self.assertEqual(self.n2.shape[1:], (50, 50, 32))
    out = self.static_2d(self.x2, is_train=True)
    out = self.dynamic_2d(self.x2, is_train=True)
    out = self.dynamic_2d(self.x2, is_train=False)
def test_BatchNorm2d(self):
    self.assertEqual(self.n2.shape[1:], (50, 50, 32))
    out = self.static_2d(self.x2, is_train=True)
    out = self.dynamic_2d(self.x2, is_train=True)
    out = self.dynamic_2d(self.x2, is_train=False)