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

!47314 [ST][MS][OPS] Fix API issues. 

Merge pull request !47314 from alashkari/fold-api-update
This commit is contained in:
i-robot 2023-01-05 16:26:24 +00:00 committed by Gitee
parent 3c77dae25d d054809185
commit b6a798329c
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/python/mindspore/common/tensor.py
View File

@ -3480,7 +3480,7 @@ class Tensor(Tensor_):
[30. 27.]
"""
self._init_check()
return tensor_operator_registry.get('addmv')(self, mat, vec, beta=1, alpha=1)
return tensor_operator_registry.get('addmv')(self, mat, vec, beta=beta, alpha=alpha)
def asinh(self):
r"""

4
mindspore/python/mindspore/ops/function/array_func.py
View File

@ -5710,11 +5710,11 @@ def diagonal(input, offset=0, dim1=0, dim2=1):
elif offset != 0:
e = e.astype(mstype.float32)
if offset > 0:
e_left = fill_op(dtype, (n, offset), 0)
e_left = fill_op(mstype.float32, (n, offset), 0)
e_right = e[..., 0:m - offset:1]
e = _get_cache_prim(P.Concat)(1)((e_left, e_right)).astype(dtype)
elif offset < 0:
e_upper = fill_op(dtype, (-offset, m), 0)
e_upper = fill_op(mstype.float32, (-offset, m), 0)
e_lower = e[0:n + offset:1, ...]
e = _get_cache_prim(P.Concat)(0)((e_upper, e_lower)).astype(dtype)
e = _get_cache_prim(P.BroadcastTo)(x_shape)(e)