パブリック最終クラスFusedBatchNormGradV3
バッチ正規化のための勾配。
4D Tensor のサイズは「NHWC」または「NCHW」のいずれかによって定義されることに注意してください。 1D テンソルのサイズは、4D テンソルの次元 C と一致します。
ネストされたクラス
クラス | FusedBatchNormGradV3.Options | FusedBatchNormGradV3 のオプションの属性 |
パブリックメソッド
static <T extends Number, U extends Number> FusedBatchNormGradV3 <T, U> | |
静的FusedBatchNormGradV3.Options | dataFormat (文字列データ形式) |
静的FusedBatchNormGradV3.Options | イプシロン(浮動小数点イプシロン) |
静的FusedBatchNormGradV3.Options | isTraining (ブール値 isTraining) |
出力<U> | offsetBackprop () オフセットに関する勾配の 1D テンソル。 |
出力<U> | リザーブスペース4 () FusedBatchNorm の平均入力と一致する未使用のプレースホルダー。 |
出力<U> | リザーブスペース5 () FusedBatchNorm の分散入力と一致する未使用のプレースホルダー。 |
出力<U> | スケールバックプロップ() スケールに関する勾配の 1D テンソル。 |
出力<T> | xBackprop () x に関する勾配の 4D テンソル。 |
継承されたメソッド
パブリックメソッド
public static FusedBatchNormGradV3 <T, U> create (スコープscope、オペランド<T> yBackprop、オペランド<T> x、オペランド<Float>スケール、オペランド<U>reserveSpace1、オペランド<U>reserveSpace2、オペランド<U>reserveSpace3、オプション...オプション)
新しい FusedBatchNormGradV3 オペレーションをラップするクラスを作成するためのファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
yバックプロップ | y に関する勾配の 4D テンソル。 |
バツ | 入力データの 4D Tensor。 |
規模 | 正規化された x をスケーリングするためのスケーリング係数の 1D テンソル。 |
予約スペース1 | is_training が True の場合、計算されたバッチの 1D テンソルは勾配計算で再利用されることを意味します。 is_training が False の場合、母集団平均の 1D テンソルは 1 次と 2 次の両方の勾配計算で再利用されます。 |
リザーブスペース2 | is_training が True の場合、勾配計算で再利用される計算されたバッチ分散 (cuDNN の場合は逆分散) の 1D テンソル。 is_training が False の場合、母分散の 1D テンソルは 1 次と 2 次の両方の勾配計算で再利用されます。 |
予約スペース3 | is_training が True の場合、一部の中間結果の 1D テンソルが勾配計算で再利用されます。 is_training が False の場合、ダミーの空の Tensor が作成されます。 |
オプション | オプションの属性値を持ちます |
戻り値
- FusedBatchNormGradV3 の新しいインスタンス
public static FusedBatchNormGradV3.Options dataFormat (String dataFormat)
パラメーター
データ形式 | y_backprop、x、x_backprop のデータ形式。 「NHWC」(デフォルト)または「NCHW」のいずれか。 |
---|
public static FusedBatchNormGradV3.Options epsilon (Float epsilon)
パラメーター
イプシロン | 小さな浮動小数点数が x の分散に追加されます。 |
---|
public static FusedBatchNormGradV3.Options isTraining (Boolean isTraining)
パラメーター
トレーニング中です | 操作がトレーニング (デフォルト) または推論のためであることを示すブール値。 |
---|