Method register_hook

knn_utils/saveEmbedMulti.py:422–424

(self, layer, func, pre=False)

        return output

    def register_hook(self, layer, func, pre=False):
        handle = layer.register_forward_pre_hook(func) if pre else layer.register_forward_hook(func)
        self.hook_handles.append(handle)
    
    def break_out(self):
        # Close the Arrow writer (main process only)