LogitGrad
产品支持情况
| 产品 |
是否支持 |
| Ascend 950PR/Ascend 950DT |
√ |
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 |
√ |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 |
√ |
| Atlas 200I/500 A2 推理产品 |
× |
| Atlas 推理系列产品 |
× |
| Atlas 训练系列产品 |
× |
功能说明
dxi={NaN,if x<0 or x>1,eps<00,if x<eps or x>1−eps,eps≥0dyixi⋅(1−xi),if eps≤xi≤1−epsdx_i=
\begin{cases}
NaN, & \text{if } x < \text0 \text{ or } x > 1 ,eps <0 \\
0, & \text{if } x < \text{eps} \text{ or } x > 1 - \text{eps},eps \geq 0 \\
\frac{dy_i}{x_i \cdot (1 - x_i)}, & \text{if } \text{eps} \leq x_i \leq 1 - \text{eps} \\
\end{cases}
dxi=⎩⎪⎪⎨⎪⎪⎧NaN,0,xi⋅(1−xi)dyi,if x<0 or x>1,eps<0if x<eps or x>1−eps,eps≥0if eps≤xi≤1−eps
参数说明
| 参数名 |
输入/输出/属性 |
描述 |
数据类型 |
数据格式 |
| x |
输入 |
公式中的x。 |
BFLOAT16、FLOAT16、FLOAT |
ND |
| dy |
输入 |
公式中的dy。 |
BFLOAT16、FLOAT16、FLOAT |
ND |
| eps |
属性 |
- 输入x的eps限制边界,防止除0错误。
- 建议值-1。
|
DOUBLE |
- |
| out |
输出 |
公式中的dx。 |
BFLOAT16、FLOAT16、FLOAT |
ND |
约束说明
- eps的取值对x和输出的影响:
- eps小于0时,x取值范围不在[0, 1],输出NaN。
- eps大于等于0时,x取值范围不在[eps, 1 - eps],输出0。
- eps大于1时,输出NaN,取值为1时为Inf。
- eps取值为Inf时,输出0。
- eps取值为NaN时,输出NaN。
- x取值对输出的影响:
- x取值0和1时,输出Inf。
- x为Inf或NaN时,输出为NaN。
- dy为Inf或NaN时,输出为Inf或NaN。
调用说明