forked from mindspore-Ecosystem/mindspore
!12237 Fix api comments of ReduceMean.
From: @zhang_yi2020 Reviewed-by: @gemini524,@liangchenghui,@ljl0711 Signed-off-by: @liangchenghui
This commit is contained in:
commit
3a4d575844
|
@ -366,9 +366,9 @@ class _Reduce(PrimitiveWithInfer):
|
|||
|
||||
class ReduceMean(_Reduce):
|
||||
"""
|
||||
Reduces a dimension of a tensor by averaging all elements in the dimension.
|
||||
Reduces a dimension of a tensor by averaging all elements in the dimension.
|
||||
|
||||
The dtype of the tensor to be reduced is number.
|
||||
The dtype of the tensor to be reduced is number.
|
||||
|
||||
Args:
|
||||
keep_dims (bool): If true, keep these reduced dimensions and the length is 1.
|
||||
|
|
Loading…
Reference in New Issue