大佬，请问是实现batchnorm1d还是batchnorm2d呢