Error: "The Muls Operator Does Not Support INT64"

Change label_batch.npu() in the preceding figure to label_batch.int().npu(). That is, change the variable type of the error line to INT32 to avoid this error.