changed floor division

supr/utils.py

@@ -48,7 +48,7 @@ def unravel_indices(indices: torch.LongTensor, shape: Tuple[int, ...]) -> torch.
     coord = []
     for dim in reversed(shape):
         coord.append(indices % dim)
-        indices = indices // dim
+        indices = torch.div(indices, dim, rounding_mode='floor')
     coord = torch.stack(coord[::-1], dim=-1)
     return coord