FrozenBatchNorm2d

class torchvision.ops.FrozenBatchNorm2d(num_features: int, eps: float = 1e-05)

BatchNorm2d，其中批次統計數據和仿射參數是固定的

參數:

• num_features (int) – 來自預期大小為 (N, C, H, W) 的輸入的特徵數量 C

• eps (float) – 為數值穩定性添加到分母的值。預設值：1e-5

forward(x: Tensor) → Tensor

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

應由所有子類別覆寫。

注意

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