LazyBatchNorm3d

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

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

延遲初始化用於根據 input.size(1) 推斷出的 BatchNorm3d 的 num_features 引數。將被延遲初始化的屬性包括 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 時，此模組跟蹤執行平均值和方差；當設為 False 時，此模組不跟蹤此類統計資訊，並將統計緩衝區 running_mean 和 running_var 初始化為 None。當這些緩衝區為 None 時，此模組在訓練和評估模式下始終使用批次統計資訊。預設值：True

cls_to_become

BatchNorm3d 的別名