Warning: This API is deprecated and will be removed in a future version of TensorFlow after the replacement is stable.

FusedBatchNormV3

public final class FusedBatchNormV3

Batch normalization.

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.

Nested Classes

class FusedBatchNormV3.Options Optional attributes for FusedBatchNormV3  

Public Methods

Output<U>
batchMean()
A 1D Tensor for the computed batch mean, to be used by TensorFlow to compute the running mean.
Output<U>
batchVariance()
A 1D Tensor for the computed batch variance, to be used by TensorFlow to compute the running variance.
static <T extends Number, U extends Number> FusedBatchNormV3<T, U>
create(Scope scope, Operand<T> x, Operand<U> scale, Operand<U> offset, Operand<U> mean, Operand<U> variance, Options... options)
Factory method to create a class wrapping a new FusedBatchNormV3 operation.
static FusedBatchNormV3.Options
dataFormat(String dataFormat)
static FusedBatchNormV3.Options
epsilon(Float epsilon)
static FusedBatchNormV3.Options
exponentialAvgFactor(Float exponentialAvgFactor)
static FusedBatchNormV3.Options
isTraining(Boolean isTraining)
Output<U>
reserveSpace1()
A 1D Tensor for the computed batch mean, to be reused in the gradient computation.
Output<U>
reserveSpace2()
A 1D Tensor for the computed batch variance (inverted variance in the cuDNN case), to be reused in the gradient computation.
Output<U>
reserveSpace3()
A 1D Tensor for some intermediate results, to be reused in the gradient computation for better efficiency.
Output<T>
y()
A 4D Tensor for output data.

Inherited Methods

Public Methods

public Output<U> batchMean ()

A 1D Tensor for the computed batch mean, to be used by TensorFlow to compute the running mean.

public Output<U> batchVariance ()

A 1D Tensor for the computed batch variance, to be used by TensorFlow to compute the running variance.

public static FusedBatchNormV3<T, U> create (Scope scope, Operand<T> x, Operand<U> scale, Operand<U> offset, Operand<U> mean, Operand<U> variance, Options... options)

Factory method to create a class wrapping a new FusedBatchNormV3 operation.

Parameters
scope current scope
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.
options carries optional attributes values
Returns
  • a new instance of FusedBatchNormV3

public static FusedBatchNormV3.Options dataFormat (String dataFormat)

Parameters
dataFormat The data format for x and y. Either "NHWC" (default) or "NCHW".

public static FusedBatchNormV3.Options epsilon (Float epsilon)

Parameters
epsilon A small float number added to the variance of x.

public static FusedBatchNormV3.Options exponentialAvgFactor (Float exponentialAvgFactor)

public static FusedBatchNormV3.Options isTraining (Boolean isTraining)

Parameters
isTraining A bool value to indicate the operation is for training (default) or inference.

public Output<U> reserveSpace1 ()

A 1D Tensor for the computed batch mean, to be reused in the gradient computation.

public Output<U> reserveSpace2 ()

A 1D Tensor for the computed batch variance (inverted variance in the cuDNN case), to be reused in the gradient computation.

public Output<U> reserveSpace3 ()

A 1D Tensor for some intermediate results, to be reused in the gradient computation for better efficiency.

public Output<T> y ()

A 4D Tensor for output data.