Closed sen-28 closed 1 year ago
Indices should be either on cpu or on the same device as the indexed tensor
Indices should be either on cpu or on the same device as the indexed tensor