Browse Source

improve docstring

tags/v1.1.0
Bairong 5 years ago
parent
commit
dcd5b1df28
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/common/tensor.py

+ 1
- 1
mindspore/common/tensor.py View File

@@ -341,7 +341,7 @@ class Tensor(Tensor_):
Reduce a dimension of a tensor by averaging all elements in the dimension.

Args:
axis (Union[None, int, tuple(int)]): Dimensions of reduction,
axis (Union[None, int, tuple(int), list(int)]): Dimensions of reduction,
when axis is None or empty tuple, reduce all dimensions.
Default: (), reduce all dimensions.
keep_dims (bool): Whether to keep the reduced dimensions.


Loading…
Cancel
Save