torch.nn.modules.module.register_module_buffer_registration_hook

torch.nn.modules.module.register_module_buffer_registration_hook(hook)

註冊一個適用於所有模組的緩衝區註冊鉤子。

警告

這會向 nn.Module 模組新增全域性狀態

每當呼叫 register_buffer() 時，都會呼叫該鉤子。它應該具有以下簽名

hook(module, name, buffer) -> None or new buffer

鉤子可以修改輸入或返回一個修改後的值。

返回

一個控制代碼，可用於透過呼叫 handle.remove() 來移除已新增的鉤子

返回型別

torch.utils.hooks.RemovableHandle