From 7873d7bc087454e3ef05c6685d88ec8c0fa203c9 Mon Sep 17 00:00:00 2001 From: root Date: Mon, 10 Aug 2020 02:48:53 +0000 Subject: [PATCH] Update the shape mess --- doc/fluid/api_cn/nn_cn/loss_cn/MarginRankingLoss.rst | 10 +++++----- 1 file changed, 5 insertions(+), 5 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 8a3cd34bb..f6125c34c 100644 --- a/doc/fluid/api_cn/nn_cn/loss_cn/MarginRankingLoss.rst +++ b/doc/fluid/api_cn/nn_cn/loss_cn/MarginRankingLoss.rst @@ -32,11 +32,11 @@ MarginRankingLoss 如果设置为 ``'mean'`` ，则返回 ``margin_rank_loss`` 的平均值。 默认值为 ``'none'`` 。 形状： - - **x** - :math:`(N, D)` ，其中D是向量的维度，数据类型为float32或float64。 - - **y** - :math:`(N, D)` ，与 ``x`` 的形状、数据类型相同。 - - **label** - :math:`(N, D)` ，与 ``x`` 的形状、数据类型相同。 - - **out** - :math:`(N)` ，如果 :attr:`reduction` 为 ``'sum'`` 或者是 ``'mean'`` ，则形状为 :math:`(1)` , - 否则shape和输入 `x` 保持一致 。数据类型与 ``x``、 ``y`` 相同。 + - **x** - :math:`(N, *)` ，维度是[N, *], 其中N 是batch size，`*` 是任意数量的额外维度，数据类型为float32或float64。 + - **y** - :math:`(N, *)` ，与 ``x`` 的形状、数据类型相同。 + - **label** - :math:`(N, *)` ，与 ``x`` 的形状、数据类型相同。 + - **out** - 如果 :attr:`reduction` 为 ``'sum'`` 或者是 ``'mean'`` ，则形状为 :math:`(1)` ， + 否则shape和输入 `x` 保持一致 。数据类型与 ``x``、 ``y`` 相同。 返回：返回计算MarginRankingLoss的可调用对象。 -- GitLab