def test_BatchNorm1d(self):
    self.assertEqual(self.n1.shape[1:], (50, 32))
    out = self.static_1d(self.x1, is_train=True)
    out = self.dynamic_1d(self.x1, is_train=True)
def test_BatchNorm1d(self):
    self.assertEqual(self.n1.shape[1:], (50, 32))
    out = self.static_1d(self.x1, is_train=True)
    out = self.dynamic_1d(self.x1, is_train=True)