2019年8月4日
[torch] pytorch hook学习
摘要： pytorch hook学习 register_hook 这里的o和z都是中间变量，不是通过指定值来定义的变量，所以是中间变量，所以pytorch并不存储这些变量的梯度。 对于中间变量z，hook的使用方式为: z.register_hook(hook_fn),其中 为一个用户自定义的函数，其签名为 阅读全文
posted @ 2019-08-04 02:52 aoru45 阅读(763) 评论(0) 推荐(0) 编辑