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

[ashawkey]

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!