LazyBatchNorm3d

class torch.nn.LazyBatchNorm3d(eps=1e-05, momentum=0.1, affine=True, track_running_stats=True, device=None, dtype=None)

一個具有延遲初始化的 torch.nn.BatchNorm3d 模組。

num_features 參數的延遲初始化是針對 BatchNorm3d 進行的，該參數是從 input.size(1) 推斷而來。 將會延遲初始化的屬性是 weight、bias、running_mean 和 running_var。

有關延遲模組及其限制的更多文件，請查看 torch.nn.modules.lazy.LazyModuleMixin。

參數

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

• momentum (Optional[float]) – 用於 running_mean 和 running_var 計算的值。 可以設定為 None 以進行累積移動平均（即簡單平均）。 預設值：0.1

• affine (bool) – 一個布林值，當設定為 True 時，此模組具有可學習的仿射參數。 預設值：True

• track_running_stats (bool) – 一個布林值，當設定為 True 時，此模組追蹤 running mean 和 variance，當設定為 False 時，此模組不追蹤此類統計資料，並將統計資料緩衝區 running_mean 和 running_var 初始化為 None。 當這些緩衝區為 None 時，此模組始終使用批次統計資料。 在訓練和評估模式中皆是如此。 預設值：True

cls_to_become

BatchNorm3d 的別名