提交 2f9ed97e 编写于 作者: Q qiaolongfei

follow comment

上级 62bf672e
...@@ -55,8 +55,6 @@ __all__ = [ ...@@ -55,8 +55,6 @@ __all__ = [
def split_lod_tensor(input, mask, level=0): def split_lod_tensor(input, mask, level=0):
""" """
**split_lod_tensor**
This function takes in an input that contains the complete lod information, This function takes in an input that contains the complete lod information,
and takes in a mask which is used to mask certain parts of the input. and takes in a mask which is used to mask certain parts of the input.
The output is the true branch and the false branch with the mask applied to The output is the true branch and the false branch with the mask applied to
......
...@@ -1638,23 +1638,23 @@ def batch_norm(input, ...@@ -1638,23 +1638,23 @@ def batch_norm(input,
Args: Args:
input(variable): The input variable which is a LoDTensor. input(variable): The input variable which is a LoDTensor.
act(string, default None): Activation type, linear|relu|prelu|... act(string, Default None): Activation type, linear|relu|prelu|...
is_test(bool, default False): Used for training or training. is_test(bool, Default False): Used for training or training.
momentum(float, default 0.9): momentum(float, Default 0.9):
epsilon(float, default 1e-05): epsilon(float, Default 1e-05):
param_attr(ParamAttr): The parameter attribute for Parameter `scale`. param_attr(ParamAttr): The parameter attribute for Parameter `scale`.
bias_attr(ParamAttr): The parameter attribute for Parameter `bias`. bias_attr(ParamAttr): The parameter attribute for Parameter `bias`.
data_layout(string, default NCHW): NCHW|NHWC data_layout(string, default NCHW): NCHW|NHWC
in_place(bool, default False): Make the input and output of batch norm reuse memory. in_place(bool, Default False): Make the input and output of batch norm reuse memory.
use_mkldnn(bool, Default false): ${use_mkldnn_comment} use_mkldnn(bool, Default false): ${use_mkldnn_comment}
name(string, Default None): A name for this layer(optional). If set None, the layer name(string, Default None): A name for this layer(optional). If set None, the layer
will be named automatically. will be named automatically.
moving_mean_name(string, Default None): The name of moving_mean which store the global Mean. moving_mean_name(string, Default None): The name of moving_mean which store the global Mean.
moving_variance_name(string, Default None): The name of the moving_variance which store the global Variance. moving_variance_name(string, Default None): The name of the moving_variance which store the global Variance.
do_model_average_for_mean_and_var(bool, Default False): do_model_average_for_mean_and_var(bool, Default False): Do model average for mean and variance or not.
Returns: Returns:
The sequence's last step variable which is a Tensor. Variable: A tensor variable which is the result after applying batch normalization on the input.
Examples: Examples:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册