!37888 fix ops.logsumexp example

Merge pull request !37888 from 吕昱峰(Nate.River)/code_docs_master
This commit is contained in:
i-robot 2022-07-13 02:40:55 +00:00 committed by Gitee
commit a678dc132f
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 1 additions and 2 deletions

View File

@ -3804,8 +3804,7 @@ def logsumexp(x, axis, keep_dims=False):
Examples:
>>> x = Tensor(np.random.randn(3, 4, 5, 6).astype(np.float32))
>>> op = ops.logsumexp(x, 1, keep_dims=True)
>>> output = op(x)
>>> output = ops.logsumexp(x, 1, keep_dims=True)
>>> print(output.shape)
(3, 1, 5, 6)
"""