Libtorch Register_hook how to use?

(kerry.cho) #1

Hi.
How to the register_hook function in Libtorch? Or do similar functions exist?