Skip to content

Commit

Permalink
PaddlePaddle Hackathon 4 No.49】:为 Paddle bce_loss 支持 float16 数据类型 (Pa…
Browse files Browse the repository at this point in the history
…ddlePaddle#5704)

* upd label_smooth fp16 dtype

* update label_smooth cn doc

* bce_loss fp16

* restore files

* restore file
  • Loading branch information
thunder95 authored Apr 17, 2023
1 parent 63ddc72 commit 40251b0
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions docs/api/paddle/nn/functional/binary_cross_entropy_cn.rst
Original file line number Diff line number Diff line change
Expand Up @@ -35,9 +35,9 @@ binary_cross_entropy

参数
:::::::::
- **input** (Tensor) - :math:`[N, *]`,其中 N 是 batch_size, `*` 是任意其他维度。输入数据 ``input`` 一般是 ``sigmoid`` 的输出。数据类型是 float32、float64。
- **input** (Tensor) - :math:`[N, *]`,其中 N 是 batch_size, `*` 是任意其他维度。输入数据 ``input`` 一般是 ``sigmoid`` 的输出。数据类型是 float16、float32、float64。
- **label** (Tensor) - :math:`[N, *]`,标签 ``label`` 的维度、数据类型与输入 ``input`` 相同。
- **weight** (Tensor,可选) - 手动指定每个 batch 二值交叉熵的权重,如果指定的话,维度必须是一个 batch 的数据的维度。数据类型是 float32, float64。默认值是:None。
- **weight** (Tensor,可选) - 手动指定每个 batch 二值交叉熵的权重,如果指定的话,维度必须是一个 batch 的数据的维度。数据类型是 float16、float32, float64。默认值是:None。
- **reduction** (str,可选) - 指定应用于输出结果的计算方式,可选值有:``'none'``, ``'mean'``, ``'sum'``。默认为 ``'mean'``,计算 `BCELoss` 的均值;设置为 ``'sum'`` 时,计算 `BCELoss` 的总和;设置为 ``'none'`` 时,则返回 bce_loss。
- **name** (str,可选) - 具体用法请参见 :ref:`api_guide_Name`,一般无需设置,默认值为 None。

Expand Down

0 comments on commit 40251b0

Please sign in to comment.