R/callbacks_hook.R
hook_output.Rd
Return a `Hook` that stores activations of `module` in `self$stored`
hook_output(module, detach = TRUE, cpu = FALSE, grad = FALSE)
module
detach or not
cpu or not
grad or not
None