pytorch如何先初始化變量,然后再賦值


下面是定義初始化

        #初始化輸入的張量 - 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

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM