pytorch 中的Variable一般常用的使用方法


Variable一般的初始化方法,默認是不求梯度的

import torch
from torch.autograd import Variable
x_tensor = torch.randn(2,3)

#將tensor轉換成Variable
x = Variable(x_tensor) 
print(x.requires_grad) #False
x = Variable(x_tensor,requires_grad=True) #Varibale 默認時不要求梯度的,如果要求梯度,需要說明
print(x)

 

 

 


免責聲明!

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



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