mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!45273 [MindSpore]change csr_mm中文注释修改 

Merge pull request !45273 from 十六夜/master
This commit is contained in:
i-robot 2022-11-10 01:27:44 +00:00 committed by Gitee
parent 4d971a11f6 75247267f9
commit 7822eb4216
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
docs/api/api_python/ops/mindspore.ops.func_csr_mm.rst
View File

@ -1,7 +1,7 @@
mindspore.ops.csr_mm
=================================
.. py:function:: mindspore.ops.csr_mm(a: CSRTensor, b: CSRTensor or Tensor, trans_a: bool, trans_b: bool, adjoint_a: bool, adjoint_b: bool)
.. py:function:: mindspore.ops.csr_mm(a: CSRTensor, b: CSRTensor, trans_a: bool, trans_b: bool, adjoint_a: bool, adjoint_b: bool)
返回稀疏矩阵a与稀疏矩阵或稠密矩阵b的矩阵乘法结果。
@ -11,11 +11,11 @@ mindspore.ops.csr_mm
参数：
- **a** (CSRTensor) - 稀疏的 CSRTensor。
- **b** (CSRTensor 或 Tensor) - 稀疏的 CSRTensor或稠密矩阵。
- **trans_a** (Tensor) - 是否对矩阵a进行转置。
- **trans_b** (Tensor) - 是否对矩阵b进行转置。
- **adjoint_a** (Tensor) - 是否对矩阵a进行共轭。
- **adjoint_b** (Tensor) - 是否对矩阵b进行共轭。
- **b** (CSRTensor) - 稀疏的 CSRTensor或稠密矩阵。
- **trans_a** (bool) - 是否对矩阵a进行转置。
- **trans_b** (bool) - 是否对矩阵b进行转置。
- **adjoint_a** (bool) - 是否对矩阵a进行共轭。
- **adjoint_b** (bool) - 是否对矩阵b进行共轭。
返回：
返回稀疏矩阵，类型为CSRTensor。

2
mindspore/python/mindspore/ops/function/sparse_func.py
View File

@ -243,7 +243,7 @@ def csr_mm(a: CSRTensor, b: CSRTensor, trans_a: bool = False, trans_b: bool = Fa
a (CSRTensor): Sparse CSR Tensor, rank should be 2.
b (CSRTensor): Sparse CSR Tensor, rank should be 2.
trans_a (bool): whether to transpose CSRTensor a.
trans_a (bool): whether to transpose CSRTensor b.
trans_b (bool): whether to transpose CSRTensor b.
adjoint_a (bool): whether to adjoint CSRTensor a.
adjoint_b (bool): whether to adjoint CSRTensor b.