public class Convolution1DLayer extends ConvolutionLayer
Modifier and Type | Class and Description |
---|---|
static class |
Convolution1DLayer.Builder |
ConvolutionLayer.AlgoMode, ConvolutionLayer.BaseConvBuilder<T extends ConvolutionLayer.BaseConvBuilder<T>>
convolutionMode, cudnnAlgoMode, kernelSize, padding, stride
nIn, nOut
activationFn, adamMeanDecay, adamVarDecay, biasInit, biasLearningRate, dist, dropOut, epsilon, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, layerName, learningRate, learningRateSchedule, momentum, momentumSchedule, rho, rmsDecay, updater, weightInit
Modifier and Type | Method and Description |
---|---|
InputType |
getOutputType(int layerIndex,
InputType inputType)
For a given type of input to this layer, what is the type of the output?
|
InputPreProcessor |
getPreProcessorForInputType(InputType inputType)
For the given type of input to this layer, what preprocessor (if any) is required?
Returns null if no preprocessor is required, otherwise returns an appropriate InputPreProcessor
for this layer, such as a CnnToFeedForwardPreProcessor |
Layer |
instantiate(NeuralNetConfiguration conf,
java.util.Collection<IterationListener> iterationListeners,
int layerIndex,
org.nd4j.linalg.api.ndarray.INDArray layerParamsView,
boolean initializeParams) |
void |
setNIn(InputType inputType,
boolean override)
Set the nIn value (number of inputs, or input depth for CNNs) based on the given input type
|
clone, getL1ByParam, getL2ByParam, getLearningRateByParam, initializer
getUpdaterByParam, resetLayerDefaultConfig
public Layer instantiate(NeuralNetConfiguration conf, java.util.Collection<IterationListener> iterationListeners, int layerIndex, org.nd4j.linalg.api.ndarray.INDArray layerParamsView, boolean initializeParams)
instantiate
in class ConvolutionLayer
public InputType getOutputType(int layerIndex, InputType inputType)
Layer
getOutputType
in class ConvolutionLayer
layerIndex
- Index of the layerinputType
- Type of input for the layerpublic void setNIn(InputType inputType, boolean override)
Layer
setNIn
in class ConvolutionLayer
inputType
- Input type for this layeroverride
- If false: only set the nIn value if it's not already set. If true: set it regardless of whether it's
already set or not.public InputPreProcessor getPreProcessorForInputType(InputType inputType)
Layer
InputPreProcessor
for this layer, such as a CnnToFeedForwardPreProcessor
getPreProcessorForInputType
in class ConvolutionLayer
inputType
- InputType to this layer