ir_version: 7 producer_name: "pytorch" producer_version: "CURRENT_VERSION" graph { node { input: "input" input: "target" output: "2" name: "SoftmaxCrossEntropyLoss_0" op_type: "SoftmaxCrossEntropyLoss" attribute { name: "ignore_index" i: -100 type: INT } attribute { name: "reduction" s: "mean" type: STRING } } name: "main_graph" input { name: "input" type { tensor_type { elem_type: 1 shape { dim { dim_value: 3 } dim { dim_value: 5 } dim { dim_value: 2 } dim { dim_value: 1 } } } } } input { name: "target" type { tensor_type { elem_type: 7 shape { dim { dim_value: 3 } dim { dim_value: 2 } dim { dim_value: 1 } } } } } output { name: "2" type { tensor_type { elem_type: 1 shape { } } } } } opset_import { version: 12 }