tensorflow::ops::FusedBatchNormV2

#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.

Arguments:

• 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.

Constructors and Destructors
FusedBatchNormV2(const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input offset, ::tensorflow::Input mean, ::tensorflow::Input variance)
FusedBatchNormV2(const ::tensorflow::Scope & scope, ::tensorflow::Input x, ::tensorflow::Input scale, ::tensorflow::Input offset, ::tensorflow::Input mean, ::tensorflow::Input variance, const FusedBatchNormV2::Attrs & attrs)

Public attributes
batch_mean	::tensorflow::Output
batch_variance	::tensorflow::Output
operation	Operation
reserve_space_1	::tensorflow::Output
reserve_space_2	::tensorflow::Output
y	::tensorflow::Output

Public static functions
DataFormat(StringPiece x)	Attrs
Epsilon(float x)	Attrs
IsTraining(bool x)	Attrs

Structs
tensorflow::ops::FusedBatchNormV2::Attrs	Optional attribute setters for FusedBatchNormV2.

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

y

::tensorflow::Output y

Public functions

FusedBatchNormV2

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

FusedBatchNormV2

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

Public static functions

DataFormat

Attrs DataFormat(
  StringPiece x
)

Epsilon

Attrs Epsilon(
  float x
)

IsTraining

Attrs IsTraining(
  bool x
)