fix bug in get_rank()

kormilitzin · May 8, 2019 · 04a2a46 · 04a2a46
1 parent 2a0e630
commit 04a2a46
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/texar/utils/shapes.py b/texar/utils/shapes.py
@@ -68,7 +68,7 @@ def get_rank(tensor: torch.Tensor) -> int:
         `None` if the rank cannot be determined.
     """
     if torch.is_tensor(tensor):
-        rank = len(tensor.dim())
+        rank = tensor.dim()
     else:
         array = np.asarray(tensor)
         rank = array.ndim