FusedBatchNorm

public final class FusedBatchNorm

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

FusedBatchNorm.Options

Optional attributes for FusedBatchNorm

Constants

String

OP_NAME

The name of this op, as known by TensorFlow core engine

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 TNumber, U extends TNumber> FusedBatchNorm<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 FusedBatchNorm operation.
static FusedBatchNorm.Options	dataFormat(String dataFormat)
static FusedBatchNorm.Options	epsilon(Float epsilon)
static FusedBatchNorm.Options	exponentialAvgFactor(Float exponentialAvgFactor)
static FusedBatchNorm.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

From class org.tensorflow.op.RawOp

final boolean	equals(Object obj)
final int	hashCode()
Operation	op() Return this unit of computation as a single Operation.
final String	toString()

From class java.lang.Object

boolean	equals(Object arg0)
final Class<?>	getClass()
int	hashCode()
final void	notify()
final void	notifyAll()
String	toString()
final void	wait(long arg0, int arg1)
final void	wait(long arg0)
final void	wait()

From interface org.tensorflow.op.Op

abstract ExecutionEnvironment	env() Return the execution environment this op was created in.
abstract Operation	op() Return this unit of computation as a single Operation.

Constants

public static final String OP_NAME

The name of this op, as known by TensorFlow core engine

Constant Value: "FusedBatchNormV3"

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 FusedBatchNorm<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 FusedBatchNorm 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 FusedBatchNorm

public static FusedBatchNorm.Options dataFormat (String dataFormat)

Parameters

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

public static FusedBatchNorm.Options epsilon (Float epsilon)

Parameters

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

public static FusedBatchNorm.Options exponentialAvgFactor (Float exponentialAvgFactor)

public static FusedBatchNorm.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.