public interface LocalResponseNormalizationHelper extends LayerHelper
Modifier and Type | Method and Description |
---|---|
INDArray |
activate(INDArray x,
boolean training,
double k,
double n,
double alpha,
double beta,
LayerWorkspaceMgr workspaceMgr) |
Pair<Gradient,INDArray> |
backpropGradient(INDArray input,
INDArray epsilon,
double k,
double n,
double alpha,
double beta,
LayerWorkspaceMgr workspaceMgr) |
boolean |
checkSupported(double k,
double n,
double alpha,
double beta) |
checkSupported, helperMemoryUse
boolean checkSupported(double k, double n, double alpha, double beta)
Pair<Gradient,INDArray> backpropGradient(INDArray input, INDArray epsilon, double k, double n, double alpha, double beta, LayerWorkspaceMgr workspaceMgr)
INDArray activate(INDArray x, boolean training, double k, double n, double alpha, double beta, LayerWorkspaceMgr workspaceMgr)
Copyright © 2021. All rights reserved.