fix getting values of a PyTorch sparse tensor

0a60df35 · rusty1s · 4bd6aa8f · 0a60df35
Commit 0a60df35 authored Jul 20, 2020 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torch_sparse/tensor.py torch_sparse/tensor.py +1 -1

No files found.
--- a/torch_sparse/tensor.py
+++ b/torch_sparse/tensor.py
@@ -66,7 +66,7 @@ class SparseTensor(object):

        value: Optional[torch.Tensor] = None
        if has_value:
-            value = mat._values()
+            value = mat.values()

        return SparseTensor(row=row, rowptr=None, col=col, value=value,
                            sparse_sizes=(mat.size(0), mat.size(1)),