Does the dispersed projection module support batchsize > 1?

[Charlulote]

Thanks for awesome work. It helps me to diffuse the anchor feature to $R^3$. When I try to use your method in my project with batchsize=3, it seems unsupported. Line 322 in dispersed_projection/map.py, idx.max() -> 41326 while the last idx of faces is 13776. Maybe it is because of pytorch3d?