xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix bug for api doc

This commit is contained in:
gengdongjie 2023-06-27 17:20:13 +08:00
parent 813953e7e9
commit cd8a501aad
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/python/mindspore/nn/layer/activation.py
View File

@ -157,7 +157,6 @@ class Softmin(Cell):
>>> output = softmin(x)
>>> print(output)
[0.2341 0.636 0.0862 0.01165 0.03168 ]
>>> assert(1.0 == output.sum())
"""
def __init__(self, axis=-1):
@ -196,7 +195,9 @@ class Softmax2d(Cell):
>>> softmax2d = nn.Softmax2d()
>>> output = softmax2d(x)
>>> print(output)
[[[[0.258, 0.28]], [[0.316, 0.342]], [[0.426, 0.378]]]
[[[[0.25838965 0.28001308]]
[[0.31559783 0.34200877]]
[[0.42601252 0.37797815]]]]
"""
def __init__(self):
@ -262,7 +263,6 @@ class Softmax(Cell):
>>> output = softmax(x)
>>> print(output)
[0.03168 0.01166 0.0861 0.636 0.2341 ]
>>> assert(1.0 == output.sum())
"""
def __init__(self, axis=-1):

2
mindspore/python/mindspore/ops/operations/math_ops.py
View File

@ -2562,7 +2562,7 @@ class Reciprocal(PrimitiveWithCheck):
>>> output = reciprocal(x)
>>> print(output)
[1. 0.5 0.25]
>>> x = Tensor(2.1+2j, mindspore.float32)
>>> x = Tensor(2.1+2j, mindspore.complex64)
>>> reciprocal = ops.Reciprocal()
>>> output = reciprocal(x)
>>> print(output)