def test_BatchNorm(self):
    self.assertEqual(self.n1_b.shape[1:], (50, 32))
    out = self.base_1d(self.x1, is_train=True)
    self.assertEqual(self.n2_b.shape[1:], (50, 50, 32))
    out = self.base_2d(self.x2, is_train=True)
    self.assertEqual(self.n3_b.shape[1:], (50, 50, 50, 32))
    out = self.base_3d(self.x3, is_train=True)
def test_BatchNorm(self):
    self.assertEqual(self.n1_b.shape[1:], (50, 32))
    out = self.base_1d(self.x1, is_train=True)
    self.assertEqual(self.n2_b.shape[1:], (50, 50, 32))
    out = self.base_2d(self.x2, is_train=True)
    self.assertEqual(self.n3_b.shape[1:], (50, 50, 50, 32))
    out = self.base_3d(self.x3, is_train=True)