InstanceNorm1d#
- class torch.nn.modules.instancenorm.InstanceNorm1d(num_features, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False, device=None, dtype=None)[原始碼]#
應用例項歸一化。
此操作在 2D(未批次的)或 3D(批次的)輸入上應用例項歸一化,如論文 Instance Normalization: The Missing Ingredient for Fast Stylization 所述。
均值和標準差是為小批次中的每個物件單獨按維度計算的。如果
affine為True,則 和 是長度為 C(其中 C 是輸入中的特徵或通道數)的可學習引數向量。方差使用有偏估計量計算,等同於 torch.var(input, unbiased=False)。預設情況下,此層在訓練和評估模式下均使用從輸入資料計算的例項統計資訊。
如果將
track_running_stats設定為True,則在訓練期間,此層會保持其計算出的均值和方差的執行估計,這些估計隨後在評估期間用於歸一化。執行估計使用預設的momentum(0.1) 進行保持。注意
此
momentum引數與最佳化器類中使用的動量以及傳統的動量概念不同。數學上,此處執行統計量的更新規則為 , 其中 是估計的統計量, 是新觀測值。注意
InstanceNorm1d和LayerNorm非常相似,但存在一些細微差別。InstanceNorm1d應用於多維時間序列等帶通道資料的每個通道,而LayerNorm通常應用於整個樣本,並且常用於 NLP 任務。此外,LayerNorm應用逐元素仿射變換,而InstanceNorm1d通常不應用仿射變換。- 引數
num_features (int) – 輸入的特徵或通道數
eps (float) – 為了數值穩定性新增到分母中的值。預設值:1e-5
momentum (Optional[float]) – 用於 running_mean 和 running_var 計算的值。預設為 0.1。
affine (bool) – 一個布林值,當設定為
True時,此模組具有可學習的仿射引數,其初始化方式與批歸一化相同。預設為False。track_running_stats (bool) – 一個布林值,當設定為
True時,此模組跟蹤執行均值和方差;當設定為False時,此模組不跟蹤這些統計資訊,並且在訓練和評估模式下始終使用批次統計資訊。預設為False。
- 形狀
輸入: 或
輸出: 或 (與輸入形狀相同)
示例
>>> # Without Learnable Parameters >>> m = nn.InstanceNorm1d(100) >>> # With Learnable Parameters >>> m = nn.InstanceNorm1d(100, affine=True) >>> input = torch.randn(20, 100, 40) >>> output = m(input)