lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update __init__.pyi 

polish docs of arg_reduce
This commit is contained in:
Xiang-Li Li 2022-02-24 17:11:57 +08:00 committed by GitHub
parent 0c8b2e3bb9
commit 43d3a47964
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/jittor/__init__.pyi
View File

@ -4716,7 +4716,7 @@ class Var:
>>> jt.arg_reduce(x, 'max', dim=1, keepdims=False)
[jt.Var([2 1], dtype=int32), jt.Var([5 7], dtype=int32)]
>>> jt.arg_reduce(x, 'min', dim=1, keepdims=False)
[jt.Var([1 2], dtype=int32), jt.Var([5 7], dtype=int32)]'''
[jt.Var([1 2], dtype=int32), jt.Var([2 1], dtype=int32)]'''
...
@overload
def reduce(self, op: str, dim: int, keepdims: bool=False)-> Var: ...