pytorch笔记

反向传播的过程

 

 

 

 

 

 定义向量的方法,  

 

out = self.w.mm(x)  #  mm表示向量相乘metrix multiple   1*2  与  2*1 相乘

 

 

 

item()方法更安全和推荐,因为它可以确保计算图的正确性,并且可以提供与Python标量类型的兼容性。而data属性已经被废弃,并且可能会导致错误,因此不建议使用。

 

 

要会看是几行几列

 

 

 

 

 

 

 

 

注意train_loader 是按照batch来划分的, 所以batch_index从0开始遍历

 

 

 

 

posted @ 2023-04-12 14:34  AKA铁柱  阅读(31)  评论(0)    收藏  举报