fix bug

haowen-xu · Apr 23, 2020 · aef0688 · aef0688
1 parent 5ff6c0a
commit aef0688
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorkit/backend/pytorch_/sparse.py b/tensorkit/backend/pytorch_/sparse.py
@@ -209,7 +209,7 @@ def shape(input: Tensor) -> List[int]:
 
 
 @sparse_jit
-def value_count(input: Tensor) -> List[int]:
+def value_count(input: Tensor) -> int:
     if not input.is_coalesced():
         input = input.coalesce()
     return input.indices().shape[1]