提交 9d78971d 编写于 作者: Z zhouhanqing

Some comments have been modified.

上级 3ca96844
...@@ -177,7 +177,7 @@ class ReduceProdOpMaker : public ReduceOpMaker { ...@@ -177,7 +177,7 @@ class ReduceProdOpMaker : public ReduceOpMaker {
public: public:
ReduceProdOpMaker(OpProto *proto, OpAttrChecker *op_checker) ReduceProdOpMaker(OpProto *proto, OpAttrChecker *op_checker)
: ReduceOpMaker(proto, op_checker) { : ReduceOpMaker(proto, op_checker) {
SetComment("ReduceProd", "prod"); SetComment("ReduceProd", "production");
AddComment(comment_); AddComment(comment_);
} }
}; };
......
...@@ -2215,8 +2215,8 @@ def reduce_prod(input, dim=None, keep_dim=False, name=None): ...@@ -2215,8 +2215,8 @@ def reduce_prod(input, dim=None, keep_dim=False, name=None):
keep_dim (bool|False): Whether to reserve the reduced dimension in the keep_dim (bool|False): Whether to reserve the reduced dimension in the
output Tensor. The result tensor will have one fewer dimension output Tensor. The result tensor will have one fewer dimension
than the :attr:`input` unless :attr:`keep_dim` is true. than the :attr:`input` unless :attr:`keep_dim` is true.
name(str|None): A name for this layer(optional). If set None, the layer name(str|None): A name for this layer(optional). If set None, the
will be named automatically. layer will be named automatically.
Returns: Returns:
Variable: The reduced Tensor variable. Variable: The reduced Tensor variable.
...@@ -2231,7 +2231,8 @@ def reduce_prod(input, dim=None, keep_dim=False, name=None): ...@@ -2231,7 +2231,8 @@ def reduce_prod(input, dim=None, keep_dim=False, name=None):
fluid.layers.reduce_prod(x) # [0.0002268] fluid.layers.reduce_prod(x) # [0.0002268]
fluid.layers.reduce_prod(x, dim=0) # [0.02, 0.06, 0.3, 0.63] fluid.layers.reduce_prod(x, dim=0) # [0.02, 0.06, 0.3, 0.63]
fluid.layers.reduce_prod(x, dim=-1) # [0.027, 0.0084] fluid.layers.reduce_prod(x, dim=-1) # [0.027, 0.0084]
fluid.layers.reduce_prod(x, dim=1, keep_dim=True) # [[0.027], [0.0084]] fluid.layers.reduce_prod(x, dim=1,
keep_dim=True) # [[0.027], [0.0084]]
""" """
helper = LayerHelper('reduce_prod', **locals()) helper = LayerHelper('reduce_prod', **locals())
out = helper.create_tmp_variable(dtype=helper.input_dtype()) out = helper.create_tmp_variable(dtype=helper.input_dtype())
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册