PerChannelNormObserver

class torchao.sparsity.PerChannelNormObserver(**kwargs)

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

forward(x_orig)

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

所有子類都應覆蓋此方法。

注意

雖然前向傳播的實現需要在該函式中定義，但之後應該呼叫 Module 例項而不是該函式，因為前者會執行已註冊的鉤子，而後者會默默忽略它們。