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

!47034 Correct tril/triu-indices docs 

Merge pull request !47034 from panzhihui/code_docs_tril
This commit is contained in:
i-robot 2022-12-21 02:55:48 +00:00 committed by Gitee
parent b31639b7ed 67ecabe981
commit 0515f11dc4
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -5882,8 +5882,7 @@ def tril_indices(row, col, offset=0, dtype=mstype.int64):
``GPU`` ``CPU`` ``GPU`` ``CPU``
Examples: Examples:
>>> net = ops.tril_indices(4, 3, -1, mindspore.int64) >>> output = ops.tril_indices(4, 3, -1, mindspore.int64)
>>> output = net()
>>> print(output) >>> print(output)
[[1 2 2 3 3 3] [[1 2 2 3 3 3]
[0 0 1 0 1 2]] [0 0 1 0 1 2]]
@ -5927,8 +5926,7 @@ def triu_indices(row, col, offset=0, dtype=mstype.int64):
``GPU`` ``CPU`` ``GPU`` ``CPU``
Examples: Examples:
>>> net = ops.triu_indices(5, 4, 2, mindspore.int64) >>> output = ops.triu_indices(5, 4, 2, mindspore.int64)
>>> output = net()
>>> print(output) >>> print(output)
[[0 0 1] [[0 0 1]
[2 3 3]] [2 3 3]]