Merge pull request asyml#9 from ZhitingHu/bugfix_pengzhi

fix bug in get_rank()
kormilitzin · May 8, 2019 · e01e6c6 · e01e6c6
2 parents 2a0e630 + 04a2a46
commit e01e6c6
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