[ONNX] Add test for Batchnorm (#45633)
Summary:
Add test for Batchnorm in training mode
Pull Request resolved: https://github.com/pytorch/pytorch/pull/45633
Reviewed By: VitalyFedyunin
Differential Revision: D24117026
Pulled By: bzinodev
fbshipit-source-id: 2728d8732e856390a2a00c3e8425b9c312c00650