JZTXT

register_forward_hook

torch.nn.Module.register_forward_hook使用

本文简单介绍 torch.nn.Module.register_forward_hook钩子函数的使用，简单写了一个卷积的网络，在net.conv1.register_forward_hook注册钩子函数，则会有module、输入input数据与卷积后输出数据output，重点说明module是关于 ......
register_forward_hook register forward Module torch
共1篇  :1/1页 首页上一页1下一页尾页