（beta）torch_npu.contrib.module.LabelSmoothingCrossEntropy

产品支持情况

[object Object]undefined

功能说明

使用NPU API进行LabelSmoothing Cross Entropy。

函数原型

参数说明

计算参数

• num_classes ([object Object])：用于onehot的class数量。
• smooth_factor ([object Object])：如果正在使用LabelSmoothing，请改为0.1。此参数取值范围为[0, 1]。默认值为0。

计算输入

• pred([object Object])：模型预测结果。
• target([object Object])：真实标签。

返回值说明

[object Object]

交叉熵计算结果。

调用示例