Refactor batch normalization

[ChrisWu1997]

Checklist

• [x] I've tested that my changes are compatible with the latest version of Tensorflow.
• [x] I've read the Contribution Guidelines
• [x] I've updated the documentation if necessary.

Motivation and Context

Support BatchNorm1d with 2D input.

Description

After refactoring, BatchNorm1d/BatchNorm2d/BatchNorm3d is basically no different from BatchNorm, except they are limited to certain input shape while BatchNorm works for all shape(at least 2D).