PyTorch裏面的torch.nn.Parameter()

在刷官方Tutorial的時候發現了一個用法self.v = torch.nn.Parameter(torch.FloatTensor(hidden_size)),看了官方教程裏面的解釋也是雲裏霧裏,因而在棧溢網看到了一篇解釋,並作了幾個實驗纔算徹底理解了這個函數。首先能夠把這個函數理解爲類型轉換函數,將一個不可訓練的類型Tensor轉換成能夠訓練的類型parameter並將這個parameter
相關文章
相關標籤/搜索