loss.item()

.item()方法是,取一个元素张量里面的具体元素值并返回该值,可以将一个零维张量转换成int型或者float型,在计算loss,accuracy时常用到。
比如

    print(epoch, i, loss.item())

理解:
1.item()取出张量具体位置的元素元素值
2.并且返回的是该位置元素值的高精度值
3.保持原元素类型不变;必须指定位置

你可能感兴趣的:(人工智能,python,pytorch,神经网络)