only the positional arguments given to the module. Keyword
      arguments won't be passed to the hooks and only to the
      "forward". The hook can modify the output. It can modify the
      input inplace but it will not have effect on forward since this
      is called after "forward()" is called. The hook should have the
      following signature:
     hook(module, args, output) -> None or modified output

  If "with_kwargs" is "True", the forward hook will be passed the
  "kwargs" given to the forward function and be expected to return
  the output possibly modified. The hook should have the following
  signature:

     hook(module, args, kwargs, output) -> None or modified output

  Parameters:
     * **hook** (*Callable*) -- The user defined hook to be
       registered.

     * **prepend** (*bool*) -- If "True", the provided "hook" will
       be fired before all existing "forward" hooks on this
