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

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

所有子類都應重寫此方法。

注意

儘管前向傳播的實現需要在此函式中定義，但您應該在之後呼叫 Module 例項而不是此函式，因為前者會處理註冊的鉤子，而後者則會靜默忽略它們。