How to register_hook on Conv2d?
Hi @Umair_Javaid,
Simply create a hook function and register it to the by calling register_forward_hook.
register_forward_hook