反向传播的过程
定义向量的方法,
out = self.w.mm(x) # mm表示向量相乘metrix multiple 1*2 与 2*1 相乘
item()方法更安全和推荐,因为它可以确保计算图的正确性,并且可以提供与Python标量类型的兼容性。而data属性已经被废弃,并且可能会导致错误,因此不建议使用。
item()
data
要会看是几行几列
注意train_loader 是按照batch来划分的, 所以batch_index从0开始遍历