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

#include <nn_ops.h>

バッチ正規化。

まとめ

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テンソル

コンストラクターとデストラクター

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)

パブリック属性

batch_mean
batch_variance
operation
reserve_space_1
reserve_space_2
y

パブリック静的関数

DataFormat (StringPiece x)
Epsilon (float x)
IsTraining (bool x)

構造体

tensorflow:: ops:: FusedBatchNorm:: Attrs

FusedBatchNormのオプションの属性セッター。

パブリック属性

バッチ平均

::tensorflow::Output batch_mean

バッチバリアンス

::tensorflow::Output batch_variance

手術

Operation operation

予約スペース_1

::tensorflow::Output reserve_space_1

予約スペース_2

::tensorflow::Output reserve_space_2

y

::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 IsTraining(
  bool x
)