您好，登录后才能下订单哦！
密码登录
登录注册
点击 登录注册 即表示同意《亿速云用户服务条款》
本篇内容主要讲解“torch.Tensor.register_hook()怎么使用”，感兴趣的朋友不妨来看看。本文介绍的方法操作简单快捷，实用性强。下面就让小编来带大家学习“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()怎么使用”有了更深的了解，不妨来实际操作一番吧！这里是亿速云网站，更多相关内容可以进入相关频道进行查询，关注我们，继续学习！
免责声明：本站发布的内容（图片、视频和文字）以原创、转载和分享为主，文章观点不代表本网站立场，如果涉及侵权请联系站长邮箱：is@yisu.com进行举报，并提供相关证据，一经查实，将立刻删除涉嫌侵权内容。