p80627345
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!31849 fix tril && triu to support bool 

Merge pull request !31849 from zhuzhongrui/pub_master2
This commit is contained in:
i-robot 2022-03-24 15:52:51 +00:00 committed by Gitee
parent 905d7f62f2 7901fd0809
commit 62e4589bb1
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/python/mindspore/numpy/array_creations.py
View File

@ -1242,7 +1242,7 @@ def tril(m, k=0):
m = m.astype(mstype.float32)
assist = nn_tril(m.shape, mstype.float32, k)
# MindSpore binary op do not support bool
elif dtype == mstype.Bool:
elif dtype == mstype.bool_:
m = m.astype(mstype.float32)
assist = nn_tril(m.shape, mstype.float32, k)
else:
@ -1288,6 +1288,10 @@ def triu(m, k=0):
if device_target == "Ascend":
m = m.astype(mstype.float32)
assist = nn_triu(m.shape, mstype.float32, k)
# MindSpore binary op do not support bool
elif dtype == mstype.bool_:
m = m.astype(mstype.float32)
assist = nn_triu(m.shape, mstype.float32, k)
else:
assist = nn_triu(m.shape, dtype, k)
return F.tensor_mul(assist, m).astype(dtype)