This module is directly picked from original pytorch3d repo.

https://github.com/facebookresearch/pytorch3d/tree/fc6a6b895185a502c4064fc5ea7ecb760fd1a0d6