Update contrib.py

update concat docs
This commit is contained in:
Xiang-Li Li 2022-02-24 18:18:16 +08:00 committed by GitHub
parent 43d3a47964
commit 83e4f53af5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -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")