テンソルフロー::作戦::融合バッチノルム

バッチ正規化。

まとめ

4D Tensor のサイズは「NHWC」または「NCHW」のいずれかによって定義されることに注意してください。 1D テンソルのサイズは、4D テンソルの次元 C と一致します。

引数:

• スコープ:スコープオブジェクト
• x: 入力データの 4Dテンソル。
• scale: 正規化された x をスケーリングするためのスケーリング係数の 1Dテンソル。
• offset: 正規化された x にシフトするオフセットの 1Dテンソル。
• means: 母集団平均の 1Dテンソル。推論のみに使用されます。トレーニングのために空にする必要があります。
• variance: 母集団分散の 1Dテンソル。推論のみに使用されます。トレーニングのために空にする必要があります。

オプションの属性 ( Attrsを参照):

• epsilon: x の分散に加算される小さな浮動小数点数。
• data_format: x と y のデータ形式。 「NHWC」（デフォルト）または「NCHW」のいずれか。
• is_training: 操作がトレーニング (デフォルト) または推論のためであることを示すブール値。

戻り値:

• Output y: 出力データの 4Dテンソル。
• Output batch_mean: 移動平均を計算するためにTensorFlowによって使用される、計算されたバッチ平均の1D Tensor 。
• Output batch_variance: 実行中の分散を計算するためにTensorFlowによって使用される、計算されたバッチ分散の1D Tensor 。
• Output reserve_space_1: 勾配計算で再利用される、計算されたバッチ平均の1Dテンソル。
• Output reserve_space_2: 勾配計算で再利用される、計算されたバッチ分散(cuDNNの場合は逆分散)の1Dテンソル。

パブリック属性

::tensorflow::Output batch_mean

::tensorflow::Output batch_variance

Operation operation

::tensorflow::Output reserve_space_1

::tensorflow::Output reserve_space_2

::tensorflow::Output y

 FusedBatchNorm(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input scale,
  ::tensorflow::Input offset,
  ::tensorflow::Input mean,
  ::tensorflow::Input variance
)

 FusedBatchNorm(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input scale,
  ::tensorflow::Input offset,
  ::tensorflow::Input mean,
  ::tensorflow::Input variance,
  const FusedBatchNorm::Attrs & attrs
)

Attrs DataFormat(
  StringPiece x
)

Attrs Epsilon(
  float x
)

Attrs ExponentialAvgFactor(
  float x
)

Attrs IsTraining(
  bool x
)