[人工智能]Pytorch使用记录

torch.max(input, dim)函数对dim维度求最大值，返回两个值，第一个是max value，第二个是max index。

分类问题计算accuracy：
根据预测结果列表prediction和label列表y计算:
torch.eq函数比较两个tensor，返回一个bool类型的tensor，相同的地方为True，不同的地方为False；通过float()函数将其转换成0、1的float类型；mean()函数用来求平均值，即accuracy，计算出来的是一个tensor，item()用来取出tensor的数值

acc = torch.eq(prediction, y).float().mean().item()