Could anyone show me how to recreate a BatchNorm2d layer?? Or just show me how to  use my Conv2d output as the input of My BatchNorm2d layer??
I know how to recreate a BatchNorm1d layer, but I ran into some problems when rewrite the BatchNorm2d layer.
If you have the solution, plz contact me.
Thank you very much~