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

setitem polish

This commit is contained in:
Dun Liang 2020-10-16 20:02:33 +08:00
parent db30ead79e
commit d31cdab30f
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/jittor/__init__.py
View File

@ -7,7 +7,7 @@
# This file is subject to the terms and conditions defined in
# file 'LICENSE.txt', which is part of this source code package.
# ***************************************************************
__version__ = '1.2.0.4'
__version__ = '1.2.0.5'
from . import lock
with lock.lock_scope():
from . import compiler

2
python/jittor/contrib.py
View File

@ -188,7 +188,7 @@ def setitem(x, slices, value):
if isinstance(slices, jt.Var) and slices.dtype == "bool":
mask = jt.broadcast(slices, x)
value = jt.broadcast(value, x)
return mask.ternary(value, mask)
return x.assign(mask.ternary(value, mask))
if isinstance(slices, list):
slices = tuple(slices)
return x.assign(x.setitem(slices, value))