Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
torch-sparse
Commits
79e08cfb
Commit
79e08cfb
authored
Jul 26, 2019
by
rusty1s
Browse files
typo
parent
eb4b8247
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
2 additions
and
2 deletions
+2
-2
README.md
README.md
+1
-1
torch_sparse/spspmm.py
torch_sparse/spspmm.py
+1
-1
No files found.
README.md
View file @
79e08cfb
...
...
@@ -202,7 +202,7 @@ Both input sparse matrices need to be **coalesced** (use the `coalesced` attribu
*
**m**
*(int)*
- The first dimension of first corresponding dense matrix.
*
**k**
*(int)*
- The second dimension of first corresponding dense matrix and first dimension of second corresponding dense matrix.
*
**n**
*(int)*
- The second dimension of second corresponding dense matrix.
*
**coalesced**
*(bool, optional)*
: If set to
`True`
, will coalesce both input sparse matrices (default:
`False`
)
.
*
**coalesced**
*(bool, optional)*
: If set to
`True`
, will coalesce both input sparse matrices
.
(default:
`False`
)
### Returns
...
...
torch_sparse/spspmm.py
View file @
79e08cfb
...
...
@@ -21,7 +21,7 @@ def spspmm(indexA, valueA, indexB, valueB, m, k, n, coalesced=False):
first dimension of second corresponding dense matrix.
n (int): The second dimension of second corresponding dense matrix.
coalesced (bool, optional): If set to :obj:`True`, will coalesce both
input sparse matrices (default: :obj:`False`)
.
input sparse matrices
.
(default: :obj:`False`)
:rtype: (:class:`LongTensor`, :class:`Tensor`)
"""
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment