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

!47568 Fix addbmm 

Merge pull request !47568 from shaojunsong/fix/addbmm
This commit is contained in:
i-robot 2023-01-06 04:23:36 +00:00 committed by Gitee
parent 9106d90d81 2b31fcaf44
commit 04e107ec7d
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
docs/api/api_python/ops/mindspore.ops.func_addbmm.rst
View File

@ -22,4 +22,5 @@
Tensor，和 `x` 具有相同的dtype。 Tensor，和 `x` 具有相同的dtype。
异常： 异常：
- **TypeError** - 如果 `alpha`，`beta` 不是int或者float。
- **ValueError** - 如果 `batch1`， `batch2` 不能进行批量矩阵乘法。 - **ValueError** - 如果 `batch1`， `batch2` 不能进行批量矩阵乘法。

5
mindspore/python/mindspore/ops/function/math_func.py
View File

@ -4758,6 +4758,7 @@ def addbmm(x, batch1, batch2, *, beta=1, alpha=1):
Tensor, has the same dtype as `x`. Tensor, has the same dtype as `x`.
Raises: Raises:
TypeError: If `alpha` or beta is not an int or float.
ValueError: If `batch1`, `batch2` cannot apply batch matrix multiplication. ValueError: If `batch1`, `batch2` cannot apply batch matrix multiplication.
Supported Platforms: Supported Platforms:
@ -4776,6 +4777,10 @@ def addbmm(x, batch1, batch2, *, beta=1, alpha=1):
[1285. 1377. 1469.] [1285. 1377. 1469.]
[1621. 1745. 1869.]] [1621. 1745. 1869.]]
""" """
if not isinstance(alpha, (int, float)):
raise TypeError(f"For 'addbmm', parameter 'alpha' must be an int or float, but got {type(alpha)}.")
if not isinstance(beta, (int, float)):
raise TypeError(f"For 'addbmm', parameter 'beta' must be an int or float, but got {type(beta)}.")
bmm_op = _get_cache_prim(P.BatchMatMul)() bmm_op = _get_cache_prim(P.BatchMatMul)()
bmm_res = bmm_op(batch1, batch2) bmm_res = bmm_op(batch1, batch2)
return beta * x + alpha * (bmm_res.sum(axis=0)) return beta * x + alpha * (bmm_res.sum(axis=0))