多分类问题计算准确率

import torch
from torch.nn import functional as F
logits=torch.rand(10,10)
pred=F.softmax(logits, dim=1)
pred_label=pred.argmax(dim=1)
label=torch.tensor([3,7,6,9,0,2,6,4,5,6])
correct=torch.eq(label,pred_label)
correct.sum().float().item()/len(label)