public static class ConvolutionLayerSpace.Builder extends FeedForwardLayerSpace.Builder<ConvolutionLayerSpace.Builder>
Modifier and Type | Field and Description |
---|---|
protected ParameterSpace<ConvolutionMode> |
convolutionMode |
protected ParameterSpace<int[]> |
dilation |
protected ParameterSpace<java.lang.Boolean> |
hasBias |
protected ParameterSpace<int[]> |
kernelSize |
protected ParameterSpace<int[]> |
padding |
protected ParameterSpace<int[]> |
stride |
constrainAll, constrainBias, constrainWeights, nIn, nOut
activationFunction, biasInit, biasUpdater, dist, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, updater, weightInit, weightNoise
dropOut
Constructor and Description |
---|
Builder() |
Modifier and Type | Method and Description |
---|---|
ConvolutionLayerSpace |
build() |
ConvolutionLayerSpace.Builder |
convolutionMode(ConvolutionMode convolutionMode) |
ConvolutionLayerSpace.Builder |
convolutionMode(ParameterSpace<ConvolutionMode> convolutionMode) |
ConvolutionLayerSpace.Builder |
dilation(int... dilation) |
ConvolutionLayerSpace.Builder |
dilation(ParameterSpace<int[]> dilation) |
ConvolutionLayerSpace.Builder |
hasBias(boolean hasBias) |
ConvolutionLayerSpace.Builder |
hasBias(ParameterSpace<java.lang.Boolean> hasBias) |
ConvolutionLayerSpace.Builder |
kernelSize(int... kernelSize) |
ConvolutionLayerSpace.Builder |
kernelSize(ParameterSpace<int[]> kernelSize) |
ConvolutionLayerSpace.Builder |
padding(int... padding) |
ConvolutionLayerSpace.Builder |
padding(ParameterSpace<int[]> padding) |
ConvolutionLayerSpace.Builder |
stride(int... stride) |
ConvolutionLayerSpace.Builder |
stride(ParameterSpace<int[]> stride) |
constrainAllParams, constrainAllParams, constrainBias, constrainBias, constrainWeights, constrainWeights, nIn, nIn, nOut, nOut
activation, activation, activation, activation, activationFn, biasInit, biasInit, biasUpdater, biasUpdater, dist, dist, gradientNormalization, gradientNormalization, gradientNormalizationThreshold, gradientNormalizationThreshold, l1, l1, l1Bias, l1Bias, l2, l2, l2Bias, l2Bias, updater, updater, weightInit, weightInit, weightInit
dropOut, dropOut, iDropOut
protected ParameterSpace<int[]> dilation
protected ParameterSpace<int[]> kernelSize
protected ParameterSpace<int[]> stride
protected ParameterSpace<int[]> padding
protected ParameterSpace<ConvolutionMode> convolutionMode
protected ParameterSpace<java.lang.Boolean> hasBias
public ConvolutionLayerSpace.Builder dilation(int... dilation)
public ConvolutionLayerSpace.Builder dilation(ParameterSpace<int[]> dilation)
public ConvolutionLayerSpace.Builder kernelSize(int... kernelSize)
public ConvolutionLayerSpace.Builder kernelSize(ParameterSpace<int[]> kernelSize)
public ConvolutionLayerSpace.Builder stride(int... stride)
public ConvolutionLayerSpace.Builder stride(ParameterSpace<int[]> stride)
public ConvolutionLayerSpace.Builder padding(int... padding)
public ConvolutionLayerSpace.Builder padding(ParameterSpace<int[]> padding)
public ConvolutionLayerSpace.Builder convolutionMode(ConvolutionMode convolutionMode)
public ConvolutionLayerSpace.Builder convolutionMode(ParameterSpace<ConvolutionMode> convolutionMode)
public ConvolutionLayerSpace.Builder hasBias(boolean hasBias)
public ConvolutionLayerSpace.Builder hasBias(ParameterSpace<java.lang.Boolean> hasBias)
public ConvolutionLayerSpace build()
build
in class LayerSpace.Builder<ConvolutionLayerSpace.Builder>