Closed Gofinge closed 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.
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.