org.tensorflow.op.core

Class FusedBatchNormV2.Options

java.lang.Object

org.tensorflow.op.core.FusedBatchNormV2.Options

Enclosing class:

FusedBatchNormV2<T extends Number,U extends Number>

public static class FusedBatchNormV2.Options
extends Object

Optional attributes for FusedBatchNormV2

Method Summary

Modifier and Type	Method and Description
FusedBatchNormV2.Options	dataFormat(String dataFormat)
FusedBatchNormV2.Options	epsilon(Float epsilon)
FusedBatchNormV2.Options	isTraining(Boolean isTraining)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

epsilon

public FusedBatchNormV2.Options epsilon(Float epsilon)

Parameters:

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

dataFormat

public FusedBatchNormV2.Options dataFormat(String dataFormat)

Parameters:

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

isTraining

public FusedBatchNormV2.Options isTraining(Boolean isTraining)

Parameters:

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