diff --git a/torch/_linalg_utils.py b/torch/_linalg_utils.py index fe168cb33c4b..43c8b65767e0 100644 --- a/torch/_linalg_utils.py +++ b/torch/_linalg_utils.py @@ -8,7 +8,7 @@ from torch import Tensor def is_sparse(A): - """Check if tensor A is a sparse tensor""" + """Check if tensor A is a sparse COO tensor. All other sparse storage formats (CSR, CSC, etc...) will return False.""" if isinstance(A, torch.Tensor): return A.layout == torch.sparse_coo