From f126db72e13767197796be149b0586b5c9203769 Mon Sep 17 00:00:00 2001 From: root Date: Sat, 8 Aug 2020 10:52:16 +0000 Subject: [PATCH] Update the code for the MarginRankingLoss, test=develop --- doc/fluid/api_cn/nn_cn/loss_cn/MarginRankingLoss.rst | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/doc/fluid/api_cn/nn_cn/loss_cn/MarginRankingLoss.rst b/doc/fluid/api_cn/nn_cn/loss_cn/MarginRankingLoss.rst index 7f986ba90..0c7681061 100644 --- a/doc/fluid/api_cn/nn_cn/loss_cn/MarginRankingLoss.rst +++ b/doc/fluid/api_cn/nn_cn/loss_cn/MarginRankingLoss.rst @@ -32,9 +32,10 @@ MarginRankingLoss 如果设置为 ``'mean'`` ，则返回 ``margin_rank_loss`` 的平均值。 默认值为 ``'none'`` 。 形状： - - **x** (Tensor) - :math:`(N, D)` ，其中D是向量的维度，数据类型为float32或float64。 - - **y** (Tensor) - :math:`(N, D)` ，与 ``x`` 的形状、数据类型相同。 - - **out** (Tensor) - :math:`(N)` ，如果 :attr:`keepdim` 为True，则形状为 :math:`(N, 1)` 。数据类型与 ``x``、 ``y`` 相同。 + - **x** - :math:`(N, D)` ，其中D是向量的维度，数据类型为float32或float64。 + - **y** - :math:`(N, D)` ，与 ``x`` 的形状、数据类型相同。 + - **out** - :math:`(N)` ，如果 :attr:`reduction` 为 ``'sum'`` 或者是 ``'mean'`` ，则形状为 :math:`(1)` , + 否则shape和输入 `x` 保持一致 。数据类型与 ``x``、 ``y`` 相同。 返回：返回计算MarginRankingLoss的可调用对象。 -- GitLab