... 轉化為LongTensor型別。 correct_k = correct[:1].view(-1).float().sum(0) top1_acc = correct_k.mul_(100.0 / 5) print("top1_acc:",top1_acc).
確定! 回上一頁