diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py index 7261b3009b727d6ab250191eb123aec72da8d41e..234692cb29593516e87c4af9943a5d567d4805f5 100644 --- a/python/paddle/fluid/layers/nn.py +++ b/python/paddle/fluid/layers/nn.py @@ -5830,24 +5830,25 @@ def rank_loss(label, left, right, name=None): def margin_rank_loss(label, left, right, margin=0.1, name=None): """ - Margin Rank loss layer for rank problem, which comparing left value and right value be passed in. - The rank loss can be defined as below equation: + Margin Ranking Loss Layer for ranking problem, + which compare left score and right score passed in. + The ranking loss can be defined as following equation: .. math:: rank\_loss &= max(0, -label * (left - right) + margin) Args: - label (Variable): Indicats whether left higher than (right + margin) or not. - left (Variable): rank score for left. - right (Variable): rank score for right. - margin (float): Indicates the margin to be added to right + label (Variable): Indicates whether the left is ranked higher than the right or not. + left (Variable): ranking score for left. + right (Variable): ranking score for right. + margin (float): Indicates the given margin to be added to right name (str|None): A name for this layer (optional). If set None, the layer will be named automatically. Returns: - list: The value of rank loss. + list: The Variable of ranking loss. Raises: - ValueError: Any of label, left, and right is not a variable. + ValueError: Any of label, left, and right is not a Variable. Examples: .. code-block:: python label = fluid.layers.data(name="label", shape=[4, 1], dtype="float32") @@ -5862,8 +5863,8 @@ def margin_rank_loss(label, left, right, margin=0.1, name=None): raise ValueError("The left should be a Variable") if not (isinstance(right, Variable)): raise ValueError("The right should be a Variable") - out = helper.create_tmp_variable("float32") - act = helper.create_tmp_variable("float32") + out = helper.create_tmp_variable(left.dtype) + act = helper.create_tmp_variable(left.dtype) helper.append_op( type='margin_rank_loss', inputs={"Label": label,