How to use in 3D input

I’m very impressed by your work, especially the visualizations presented in your paper. 
I have a question regarding the visualization code for 3D models. In your layers_ours.py, modules like Conv2d and Bathchnorm2d are used. Could you please advise how these might be expanded to 3D (for example, using Conv3d)