テンソルフロー::作戦::融合バッチノルム
#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
バッチ分散: 実行分散を計算するために 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のオプションの属性セッター。 |
パブリック属性
バッチ平均
::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 )