tensorflow:: ops:: FusedBatchNormV3
#include <nn_ops.h>
Batch normalization.
Summary
Note that the size of 4D Tensors are defined by either "NHWC" or "NCHW". The size of 1D Tensors matches the dimension C of the 4D Tensors.
Args:
- scope: A Scope object
- x: A 4D Tensor for input data.
- scale: A 1D Tensor for scaling factor, to scale the normalized x.
- offset: A 1D Tensor for offset, to shift to the normalized x.
- mean: A 1D Tensor for population mean. Used for inference only; must be empty for training.
- variance: A 1D Tensor for population variance. Used for inference only; must be empty for training.
Optional attributes (see Attrs
):
- epsilon: A small float number added to the variance of x.
- data_format: The data format for x and y. Either "NHWC" (default) or "NCHW".
- is_training: A bool value to indicate the operation is for training (default) or inference.
Returns:
Output
y: A 4D Tensor for output data.Output
batch_mean: A 1D Tensor for the computed batch mean, to be used by TensorFlow to compute the running mean.Output
batch_variance: A 1D Tensor for the computed batch variance, to be used by TensorFlow to compute the running variance.Output
reserve_space_1: A 1D Tensor for the computed batch mean, to be reused in the gradient computation.Output
reserve_space_2: A 1D Tensor for the computed batch variance (inverted variance in the cuDNN case), to be reused in the gradient computation.Output
reserve_space_3: A 1D Tensor for some intermediate results, to be reused in the gradient computation for better efficiency.
Constructors and Destructors |
|
---|---|
FusedBatchNormV3(const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input offset, ::tensorflow::Input mean, ::tensorflow::Input variance)
|
|
FusedBatchNormV3(const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input offset, ::tensorflow::Input mean, ::tensorflow::Input variance, const FusedBatchNormV3::Attrs & attrs)
|
Public attributes |
|
---|---|
batch_mean
|
|
batch_variance
|
|
operation
|
|
reserve_space_1
|
|
reserve_space_2
|
|
reserve_space_3
|
|
y
|
Public static functions |
|
---|---|
DataFormat(StringPiece x)
|
|
Epsilon(float x)
|
|
ExponentialAvgFactor(float x)
|
|
IsTraining(bool x)
|
Structs |
|
---|---|
tensorflow:: |
Optional attribute setters for FusedBatchNormV3. |
Public attributes
batch_mean
::tensorflow::Output batch_mean
batch_variance
::tensorflow::Output batch_variance
operation
Operation operation
reserve_space_1
::tensorflow::Output reserve_space_1
reserve_space_2
::tensorflow::Output reserve_space_2
reserve_space_3
::tensorflow::Output reserve_space_3
y
::tensorflow::Output y
Public functions
FusedBatchNormV3
FusedBatchNormV3( const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input offset, ::tensorflow::Input mean, ::tensorflow::Input variance )
FusedBatchNormV3
FusedBatchNormV3( const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input offset, ::tensorflow::Input mean, ::tensorflow::Input variance, const FusedBatchNormV3::Attrs & attrs )
Public static functions
DataFormat
Attrs DataFormat( StringPiece x )
Epsilon
Attrs Epsilon( float x )
ExponentialAvgFactor
Attrs ExponentialAvgFactor( float x )
IsTraining
Attrs IsTraining( bool x )