01 2022 档案

pytorch中对形如a[0].data进行赋值时的一个坑点

摘要：假设我们定义了 import torch a=torch(10,3) c=torch(1,3) 现在想要对a[0]赋值成c，那么当然a[0]=c就可以完成这种操作。但有时因为要隔绝梯度的原因（或者单纯就是脑抽了），可能我们会写成a[0].data=c，但这样做的话，你会发现a[0]完全没有变化，而阅读全文

posted @ 2022-01-09 18:39 炸鸡块君阅读(105) 评论(0) 推荐(0) 编辑