[BUG] `coords` batch index should be 0 instead of 3 in `bev.py`

[kiui]

Is there an existing issue for this?

• I have searched the existing issues

Current Behavior

torchsparse/torchsparse/nn/modules/bev.py

Line 96 in 462dea4

 coords = (coords - self.offset).t()[[3] + self.bev_dims].long() 

It seems some modules still use 3 as the batch index instead of 0 in the latest implementation.

Expected Behavior

Modify to coords = (coords - self.offset).t()[[0] + self.bev_dims].long()

Environment

- GCC:
- NVCC:
- PyTorch:
- PyTorch CUDA:
- TorchSparse:

Anything else?

No response

[Zhijian Liu]

Thank you for reporting this issue. Could you please submit a short pull request for this? Thanks!