PerChannelNormObserver

class torchao.sparsity.PerChannelNormObserver(**kwargs)

一個自訂的觀察器，計算每個通道的 L2 範數，並將其儲存在緩衝區中。

forward(x_orig)

定義每次呼叫時執行的計算。

應該由所有子類別覆寫。

注意

雖然 forward pass 的配方需要在這個函數中定義，但之後應該呼叫 Module 實例，而不是這個函數，因為前者會處理執行已註冊的 Hook，而後者會默默地忽略它們。