Commit eb4b8247 authored by rusty1s's avatar rusty1s
Browse files

typo

parent 31c6292a
...@@ -191,7 +191,7 @@ torch_sparse.spspmm(indexA, valueA, indexB, valueB, m, k, n) -> (torch.LongTenso ...@@ -191,7 +191,7 @@ torch_sparse.spspmm(indexA, valueA, indexB, valueB, m, k, n) -> (torch.LongTenso
``` ```
Matrix product of two sparse tensors. Matrix product of two sparse tensors.
Both input sparse matrices need to be **coalesced** (use the `coalesce` attribute to force). Both input sparse matrices need to be **coalesced** (use the `coalesced` attribute to force).
### Parameters ### Parameters
......
...@@ -9,7 +9,7 @@ if torch.cuda.is_available(): ...@@ -9,7 +9,7 @@ if torch.cuda.is_available():
def spspmm(indexA, valueA, indexB, valueB, m, k, n, coalesced=False): def spspmm(indexA, valueA, indexB, valueB, m, k, n, coalesced=False):
"""Matrix product of two sparse tensors. Both input sparse matrices need to """Matrix product of two sparse tensors. Both input sparse matrices need to
be coalesced (use the :obj:`coalesce` attribute to force). be coalesced (use the :obj:`coalesced` attribute to force).
Args: Args:
indexA (:class:`LongTensor`): The index tensor of first sparse matrix. indexA (:class:`LongTensor`): The index tensor of first sparse matrix.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment