Pytorch如何先初始化变量，然后再赋值

最新推荐文章于 2024-05-24 10:38:24 发布

开飞机的小毛驴儿

最新推荐文章于 2024-05-24 10:38:24 发布

阅读量1.6k

点赞数 1

分类专栏：深度学习

原文链接：https://www.cnblogs.com/wanghui-garcia/p/11278345.html

版权

深度学习专栏收录该内容

141 篇文章 18 订阅

订阅专栏

转载自：https://www.cnblogs.com/wanghui-garcia/p/11278345.html，本文只做个人记录学习使用，版权归原作者所有。

下面是定义初始化

#初始化输入的变量，torch.empty是返回一个包含未初始化数据的张量
self.input=torch.empty(size=(self.opt.batchsize,3,self.opt.isize,self.opt.isize),dtype=torch.float32,device=self.device)
self.label=torch.empty(size=(self.opt.batchsize,),dtype=torch.float32,device=self.device)

然后再进行动态赋值

#设置输入的数据
def set_input(self,input):
    # 把data的第一项即图片数据赋值给self.input
    self.input.data.resize_(input[0].size()).copy_(input[0]) 
    # 把data的第二项即图像的标签赋值给self.label
    self.label.data.resize_(input[1].size()).copy_(input[1])

其实直到这里，我也还是看得似懂非懂，等后面有更深入理解的时候我自己再回来补充！