Fix the dimension index

mit-han-lab · Jul 31, 2024 · 2f38072 · 2f38072
1 parent 1def50d
commit 2f38072
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchsparse/nn/modules/bev.py b/torchsparse/nn/modules/bev.py
@@ -197,7 +197,7 @@ def forward(self, input: SparseTensor) -> torch.Tensor:
         assert isinstance(stride, torch.Tensor), type(stride)
 
         # [b, x, y, z]
-        coords = (coords - self.offset).t()[[3] + self.bev_dims + [self.dim]].long()
+        coords = (coords - self.offset).t()[[0] + self.bev_dims + [self.dim]].long()
         shape = self.shape[self.bev_dims + [self.dim]]
 
         # now stride must be torch.Tensor since input.s is tuple.