torch.Tensor.register_hook()怎么使用

本篇內(nèi)容主要講解“torch.Tensor.register_hook()怎么使用”，感興趣的朋友不妨來看看。本文介紹的方法操作簡單快捷，實用性強。下面就讓小編來帶大家學(xué)習(xí)“torch.Tensor.register_hook()怎么使用”吧!

代碼實驗舉例1:

C:\Users\chenxuqi>conda activate ssd4pytorch2_2_0(ssd4pytorch2_2_0) C:\Users\chenxuqi>python
Python 3.7.7 (default, May  6 2020, 11:45:54) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.>>> import torch>>> torch.manual_seed(seed=20200910)<torch._C.Generator object at 0x000001EB8032D330>>>>>>> v = torch.tensor([0., 0., 0.], requires_grad=True)>>> h = v.register_hook(lambda grad: grad * 2)  # double the gradient>>> print(v.grad)None>>> v.backward(torch.tensor([1., 2., 3.]))>>> print(v.grad)tensor([2., 4., 6.])>>> v.grad
tensor([2., 4., 6.])>>> h.remove()  # removes the hook>>> v.grad.zero_()tensor([0., 0., 0.])>>> v.grad
tensor([0., 0., 0.])>>> v.backward(torch.tensor([1., 2., 3.]))>>> v.grad
tensor([1., 2., 3.])>>>>>>

代碼實驗舉例2:

C:\Users\chenxuqi>conda activate ssd4pytorch2_2_0(ssd4pytorch2_2_0) C:\Users\chenxuqi>python
Python 3.7.7 (default, May  6 2020, 11:45:54) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.>>> import torch>>> torch.manual_seed(seed=20200910)<torch._C.Generator object at 0x00000259BC41D330>>>>>>> v = torch.tensor([0., 0., 0.], requires_grad=True)>>> print(v.grad)None>>> v.backward(torch.tensor([1., 2., 3.]))>>> print(v.grad)tensor([1., 2., 3.])>>>>>>>>>

到此，相信大家對“torch.Tensor.register_hook()怎么使用”有了更深的了解，不妨來實際操作一番吧！這里是億速云網(wǎng)站，更多相關(guān)內(nèi)容可以進(jìn)入相關(guān)頻道進(jìn)行查詢，關(guān)注我們，繼續(xù)學(xué)習(xí)！