All Methods Instance Methods Abstract Methods
Modifier and Type |
Method and Description |
org.nd4j.linalg.api.ndarray.INDArray |
activate(org.nd4j.linalg.api.ndarray.INDArray z,
org.nd4j.linalg.activations.IActivation afn) |
Pair<Gradient,org.nd4j.linalg.api.ndarray.INDArray> |
backpropGradient(org.nd4j.linalg.api.ndarray.INDArray input,
org.nd4j.linalg.api.ndarray.INDArray weights,
org.nd4j.linalg.api.ndarray.INDArray delta,
int[] kernel,
int[] strides,
int[] pad,
org.nd4j.linalg.api.ndarray.INDArray biasGradView,
org.nd4j.linalg.api.ndarray.INDArray weightGradView,
org.nd4j.linalg.activations.IActivation afn,
ConvolutionLayer.AlgoMode mode,
ConvolutionMode convolutionMode) |
org.nd4j.linalg.api.ndarray.INDArray |
preOutput(org.nd4j.linalg.api.ndarray.INDArray input,
org.nd4j.linalg.api.ndarray.INDArray weights,
org.nd4j.linalg.api.ndarray.INDArray bias,
int[] kernel,
int[] strides,
int[] pad,
ConvolutionLayer.AlgoMode mode,
ConvolutionMode convolutionMode) |