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: D18266918
Pulled By: yf225
fbshipit-source-id: f432904c72985d52ec52cb992cceb372b6ff0244
