public interface BatchNormalizationHelper extends LayerHelper
Modifier and Type | Method and Description |
---|---|
Pair<Gradient,INDArray> |
backpropGradient(INDArray input,
INDArray epsilon,
int[] shape,
INDArray gamma,
INDArray dGammaView,
INDArray dBetaView,
double eps,
LayerWorkspaceMgr workspaceMgr) |
boolean |
checkSupported(double eps,
boolean fixedGammaBeta) |
INDArray |
getMeanCache(org.nd4j.linalg.api.buffer.DataType dataType) |
INDArray |
getVarCache(org.nd4j.linalg.api.buffer.DataType dataType) |
INDArray |
preOutput(INDArray x,
boolean training,
int[] shape,
INDArray gamma,
INDArray beta,
INDArray mean,
INDArray var,
double decay,
double eps,
LayerWorkspaceMgr workspaceMgr) |
helperMemoryUse
boolean checkSupported(double eps, boolean fixedGammaBeta)
Pair<Gradient,INDArray> backpropGradient(INDArray input, INDArray epsilon, int[] shape, INDArray gamma, INDArray dGammaView, INDArray dBetaView, double eps, LayerWorkspaceMgr workspaceMgr)
INDArray preOutput(INDArray x, boolean training, int[] shape, INDArray gamma, INDArray beta, INDArray mean, INDArray var, double decay, double eps, LayerWorkspaceMgr workspaceMgr)
INDArray getMeanCache(org.nd4j.linalg.api.buffer.DataType dataType)
INDArray getVarCache(org.nd4j.linalg.api.buffer.DataType dataType)
Copyright © 2019. All rights reserved.