文件最后提交记录最后更新时间
2 个月前
5 个月前
4 个月前
3 个月前
2 个月前
2 个月前
2 个月前
3 个月前
4 个月前
README

BinaryCrossEntropy

产品支持情况

产品 是否支持
Ascend 950PR/Ascend 950DT
Atlas A3 训练系列产品/Atlas A3 推理系列产品
Atlas A2 训练系列产品/Atlas A2 推理系列产品
Atlas 200I/500 A2 推理产品 ×
Atlas 推理系列产品
Atlas 训练系列产品

功能说明

  • 算子功能:计算x和y的二元交叉熵。

  • 计算公式:

    当reduction为None时:

    ℓ(x,y)=L={l1,...,ln}T,ℓn=−weightn[yn⋅log(xn)+(1−yn⋅log(1−xn))]\ell(x, y)= L = \{l_{1},...,l_{n}\}^{T}, \ell_{n} = - weight_{n}[y_{n}·log(x_{n}) + (1 - y_{n}·log(1-x_{n}))]

    当reduction不为None时:

    ℓ(x,y){mean(L),if reduction=meansum(L),if reduction=sum\ell(x, y) \begin{cases} mean(L), & if\ reduction = mean \\ sum(L), & if\ reduction = sum \\ \end{cases}

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 表示预测的概率值,公式中的输入`x`。 BFLOAT16、FLOAT16、FLOAT32 ND
y 输入 表示目标张量,公式中的输入`y`。 BFLOAT16、FLOAT16、FLOAT32 ND
weight 输入 表示权重张量,公式中的输入`weight`。 BFLOAT16、FLOAT16、FLOAT32 ND
output 输出 表示计算输出 BFLOAT16、FLOAT16、FLOAT32 ND
reduction 属性 表示规约方式,公式中的输入`reduction`,输出规约的枚举值。 STRING -

约束说明

无。

调用说明

调用方式 样例代码 说明
aclnn接口 test_aclnn_binary_cross_entropy.cpp 通过aclnnBinaryCrossEntropy接口方式调用BinaryCrossEntropy算子。