NVIDIA / MinkowskiEngine

Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors
https://nvidia.github.io/MinkowskiEngine
Other
2.43k stars 360 forks source link

fix bug of device detect in batched_coordinates #418

Closed Gofinge closed 2 years ago

Gofinge commented 2 years ago

coords in batched_coordinates is a sequence of tensor: List(tensor), while the device is none, we should check the device of one of the coords.