- Function: Calculates the KL divergence.
- Formulas:
Defines loss_pointwise and saves the intermediate result.
Formula for computing
[object Object]:
Each operator has calls. First, [object Object] is called to obtain the workspace size required for computation and the executor that contains the operator computation flow. Then, [object Object] is called to perform computation.
[object Object]
[object Object]
Parameter description:
[object Object]- [object Object]Atlas inference products[object Object] and [object Object]Atlas training products[object Object]: BFLOAT16 is not supported.
Returns
[object Object]: status code. For details, see .The first-phase API performs input parameter validation. The following errors may be returned:
[object Object]
- Deterministic computation:
- Default deterministic implementation of aclnnKlDiv.
The following example is for reference only. For details, see .
[object Object]