Update contrib.py

update concat docs
2022-02-24 18:18:16 +08:00 · 2022-02-24 18:18:16 +08:00 · 83e4f53af5
parent 43d3a47964
commit 83e4f53af5
1 changed files with 1 additions and 1 deletions
--- a/python/jittor/contrib.py
+++ b/python/jittor/contrib.py
@ -243,7 +243,7 @@ def concat(arr, dim=0):
 Example::

        jt.concat([jt.array([[1],[2]]), jt.array([[2],[2]])], dim=1)
-        # return [[1],[2],[2],[2]]
+        # return jt.Var([[1,2],[2,2]],dtype=int32)
    '''
    if not isinstance(arr, Sequence):
        raise TypeError("concat arr needs to be a tuple or list")