it's kind of a niche use case, but some work has been done to implement octrees for volumetric data, this saves a lot of space and can really be helpful if your data is sparse. Some code for torch is here https://github.com/griegler/octnet. Is there a way we can integrate this into pytorch?