org.tensorflow.op.core

Class FusedBatchNormV3.Options

java.lang.Object

org.tensorflow.op.core.FusedBatchNormV3.Options

Enclosing class:

FusedBatchNormV3<T extends Number,U extends Number>

public static class FusedBatchNormV3.Options
extends Object

Optional attributes for FusedBatchNormV3

Method Summary

Modifier and Type	Method and Description
FusedBatchNormV3.Options	dataFormat(String dataFormat)
FusedBatchNormV3.Options	epsilon(Float epsilon)
FusedBatchNormV3.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 FusedBatchNormV3.Options epsilon(Float epsilon)

Parameters:

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

dataFormat

public FusedBatchNormV3.Options dataFormat(String dataFormat)

Parameters:

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

isTraining

public FusedBatchNormV3.Options isTraining(Boolean isTraining)

Parameters:

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