Imported from https://github.com/lxx1991/pytorch-voxel-flow