欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

PyTorch中 tensor.detach() 和 tensor.data 的區(qū)別詳解

 更新時(shí)間:2020年01月06日 14:46:20   作者:夢(mèng)家  
今天小編就為大家分享一篇PyTorch中 tensor.detach() 和 tensor.data 的區(qū)別詳解,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧

PyTorch0.4中,.data 仍保留,但建議使用 .detach(), 區(qū)別在于 .data 返回和 x 的相同數(shù)據(jù) tensor, 但不會(huì)加入到x的計(jì)算歷史里,且require s_grad = False, 這樣有些時(shí)候是不安全的, 因?yàn)?x.data 不能被 autograd 追蹤求微分 。

.detach() 返回相同數(shù)據(jù)的 tensor ,且 requires_grad=False ,但能通過 in-place 操作報(bào)告給 autograd 在進(jìn)行反向傳播的時(shí)候.

舉例:

tensor.data

>>> a = torch.tensor([1,2,3.], requires_grad =True)
>>> out = a.sigmoid()
>>> c = out.data
>>> c.zero_()
tensor([ 0., 0., 0.])

>>> out     # out的數(shù)值被c.zero_()修改
tensor([ 0., 0., 0.])

>>> out.sum().backward() # 反向傳播
>>> a.grad    # 這個(gè)結(jié)果很嚴(yán)重的錯(cuò)誤,因?yàn)閛ut已經(jīng)改變了
tensor([ 0., 0., 0.])

tensor.detach()

>>> a = torch.tensor([1,2,3.], requires_grad =True)
>>> out = a.sigmoid()
>>> c = out.detach()
>>> c.zero_()
tensor([ 0., 0., 0.])

>>> out     # out的值被c.zero_()修改 !!
tensor([ 0., 0., 0.])

>>> out.sum().backward() # 需要原來out得值,但是已經(jīng)被c.zero_()覆蓋了,結(jié)果報(bào)錯(cuò)
RuntimeError: one of the variables needed for gradient
computation has been modified by an

以上這篇PyTorch中 tensor.detach() 和 tensor.data 的區(qū)別詳解就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。

相關(guān)文章

最新評(píng)論