C++ API: torch::nn::BatchNorm{2,3}d (#28936)
Summary:
Add torch::nn::BatchNorm{2,3}d module and functional support for the C++ API.
Related Issue: https://github.com/pytorch/pytorch/issues/25883 #28176
Reviewer: yf225
Pull Request resolved: https://github.com/pytorch/pytorch/pull/28936
Differential Revision: D18274584
Pulled By: yf225
fbshipit-source-id: 3784eee9f8947f6c7c9f1699544a3d36a1a019b7
