linear_operator
linear_operator copied to clipboard
Fix dtype/device mismatch in _get_indices()
Previously, this method failed if row_index and col_index have different dtype/device. Now, we first unify these and do the comparisons afterwards.
Differing dtype/device combinations can be created for example by _convert_indices_to_tensors() in __getitem__() (reference).