问题现象描述

Muls算子不支持INT64。

解决措施

如图所示，将label_batch.npu()改成label_batch.int().npu()，即把当前报错行的变量类型改成int32，规避此类Muls算子不支持int64的问题。