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-scatter
Commits
ecb4a9ee
Unverified
Commit
ecb4a9ee
authored
Nov 08, 2021
by
lausannel
Committed by
GitHub
Nov 07, 2021
Browse files
Change other to other.size() for compatibility (#249)
* change other to other.size() for compatibility * fix typo
parent
8d05f610
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
torch_scatter/utils.py
torch_scatter/utils.py
+1
-1
No files found.
torch_scatter/utils.py
View file @
ecb4a9ee
...
...
@@ -9,5 +9,5 @@ def broadcast(src: torch.Tensor, other: torch.Tensor, dim: int):
src
=
src
.
unsqueeze
(
0
)
for
_
in
range
(
src
.
dim
(),
other
.
dim
()):
src
=
src
.
unsqueeze
(
-
1
)
src
=
src
.
expand
_as
(
other
)
src
=
src
.
expand
(
other
.
size
()
)
return
src
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