public static class FusedBatchNorm.Options extends Object
FusedBatchNorm
Modifier and Type | Method and Description |
---|---|
FusedBatchNorm.Options |
dataFormat(String dataFormat) |
FusedBatchNorm.Options |
epsilon(Float epsilon) |
FusedBatchNorm.Options |
isTraining(Boolean isTraining) |
public FusedBatchNorm.Options epsilon(Float epsilon)
epsilon
- A small float number added to the variance of x.public FusedBatchNorm.Options dataFormat(String dataFormat)
dataFormat
- The data format for x and y. Either "NHWC" (default) or "NCHW".public FusedBatchNorm.Options isTraining(Boolean isTraining)
isTraining
- A bool value to indicate the operation is for training (default)
or inference.Copyright © 2022. All rights reserved.