Yeah, well the hook is broken. 
But this in particular means that the structure just comes from the last operation in the forward of the module.
For some modules (conv, linear that doesn’t use broadcasting), that is OK, for others not so much.
Best regards
Thomas