下面是定義初始化
#初始化輸入的張量 - 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): self.input.data.resize_(input[0].size()).copy_(input[0]) #把data的第一項:圖片數據賦值給self.input self.label.data.resize_(input[1].size()).copy_(input[1]) #把data的第二項:圖片的標簽賦值給sele.gt
