Recreate a BatchNorm2d layer

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~