tensorflow :: ops :: FusedBatchNorm

#include <nn_ops.h>

バッチ正規化。

概要

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

引数:

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

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

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

戻り値:

  • Output y:出力データ用の4Dテンソル
  • Output batch_mean:実行中の平均を計算するためにTensorFlowによって使用される、計算されたバッチ平均の1Dテンソル
  • Output batch_variance:実行中の分散を計算するためにTensorFlowによって使用される、計算されたバッチ分散の1Dテンソル
  • 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のオプションの属性セッター。

パブリック属性

batch_mean

::tensorflow::Output batch_mean

batch_variance

::tensorflow::Output batch_variance

操作

Operation operation

reserved_space_1

::tensorflow::Output reserve_space_1

reserved_space_2

::tensorflow::Output reserve_space_2

y

::tensorflow::Output y

公の行事

FusedBatchNorm

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

FusedBatchNorm

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

パブリック静的関数

DataFormat

Attrs DataFormat(
  StringPiece x
)

イプシロン

Attrs Epsilon(
  float x
)

IsTraining

Attrs IsTraining(
  bool x
)