SoftmaxCrossEntropyLoss

功能

计算Softmax交叉熵损失。

边界

【输入4-5】

scores：tensor，shape=[batch_size, class_size]或[batch_size, class_size, D1, D2 , ..., Dk]（其中K是维数），数据类型：float16、float、double、bfloat16

labels：tensor，shape=[batch_size]或[batch_size, D1, D2, ..., Dk]（其中K是维数），数据类型：int32、int64

weights：tensor，数据类型：float16、float、double、bfloat16

ignore_index：指定被忽略且不影响输入梯度的目标值，为整数，可选参数

reduction：none、mean（默认值）、sum

【输出1-2】

output：tensor（loss），当reduction取值为“none”时，在K维损失的情况下：shape=[batch_size]或[batch_size, D1, D2, ..., Dk]，数据类型：float16、float、double、bfloat16

log_prob：对数概率张量。如果Softmax的输出为prob，则其值为log(prob)。shape=[batch_size, class_size]或[batch_size, class_size, D1, D2 , ..., Dk]（其中K是维数），数据类型：float16、float、double、bfloat16

支持的ONNX版本

Opset v12/v13/v14/v15/v16/v17/v18