org.tensorflow.op.core

Class FusedBatchNorm.Options

java.lang.Object

org.tensorflow.op.core.FusedBatchNorm.Options

Enclosing class:

FusedBatchNorm<T extends Number>

public static class FusedBatchNorm.Options
extends Object

Optional attributes for FusedBatchNorm

Method Summary

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

Parameters:

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

dataFormat

public FusedBatchNorm.Options dataFormat(String dataFormat)

Parameters:

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

isTraining

public FusedBatchNorm.Options isTraining(Boolean isTraining)

Parameters:

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