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

polish getitem

This commit is contained in:
li-xl 2021-03-12 15:40:29 +08:00
parent dd6394e656
commit ae0a661c3b
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/contrib.py
View File

@ -179,7 +179,7 @@ def _setitem_old(x, slices, value):
# PATCH
def getitem(x, slices):
if isinstance(slices, jt.Var) and slices.dtype == "bool":
return getitem(x, slices.where())
return getitem(x, tuple(slices.where()))
if isinstance(slices, tuple):
ss = []
for s in slices: