快捷方式

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 例項而不是此函式本身,因為前者會負責執行已註冊的鉤子,而後者會靜默地忽略它們。

文件

獲取 PyTorch 全面的開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源