public static class Deconvolution3D.Builder extends ConvolutionLayer.BaseConvBuilder<Deconvolution3D.Builder>
convolutionDim, convolutionMode, cudnnAlgoMode, cudnnAllowFallback, cudnnBwdDataAlgo, cudnnBwdFilterAlgo, cudnnFwdAlgo, dilation, hasBias, kernelSize, padding, stride
nIn, nOut
activationFn, biasInit, biasUpdater, gainInit, gradientNormalization, gradientNormalizationThreshold, iupdater, regularization, regularizationBias, weightInitFn, weightNoise
allParamConstraints, biasConstraints, iDropout, layerName, weightConstraints
Constructor and Description |
---|
Builder() |
Modifier and Type | Method and Description |
---|---|
protected boolean |
allowCausal() |
Deconvolution3D |
build() |
Deconvolution3D.Builder |
convolutionMode(ConvolutionMode convolutionMode)
Set the convolution mode for the Convolution layer.
|
Deconvolution3D.Builder |
dataFormat(Convolution3D.DataFormat dataFormat) |
Deconvolution3D.Builder |
kernelSize(int... kernelSize)
Size of the convolution rows/columns
|
Deconvolution3D.Builder |
padding(int... padding) |
void |
setDilation(int... dilation) |
void |
setKernelSize(int... kernelSize) |
void |
setPadding(int... padding) |
void |
setStride(int... stride) |
Deconvolution3D.Builder |
stride(int... stride) |
cudnnAlgoMode, cudnnAllowFallback, cudnnBwdDataMode, cudnnBwdFilterMode, cudnnFwdMode, dilation, hasBias, helperAllowFallback, setConvolutionMode
nIn, nIn, nOut, nOut, units
activation, activation, biasInit, biasUpdater, dist, gainInit, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, regularization, regularizationBias, updater, updater, weightDecay, weightDecay, weightDecayBias, weightDecayBias, weightInit, weightInit, weightInit, weightNoise
constrainAllParameters, constrainBias, constrainWeights, dropOut, dropOut, name
protected boolean allowCausal()
allowCausal
in class ConvolutionLayer.BaseConvBuilder<Deconvolution3D.Builder>
public Deconvolution3D.Builder convolutionMode(ConvolutionMode convolutionMode)
ConvolutionMode
for more detailsconvolutionMode
in class ConvolutionLayer.BaseConvBuilder<Deconvolution3D.Builder>
convolutionMode
- Convolution mode for layerpublic Deconvolution3D.Builder kernelSize(int... kernelSize)
kernelSize
in class ConvolutionLayer.BaseConvBuilder<Deconvolution3D.Builder>
kernelSize
- the height and width of the kernelpublic Deconvolution3D.Builder stride(int... stride)
stride
in class ConvolutionLayer.BaseConvBuilder<Deconvolution3D.Builder>
public Deconvolution3D.Builder padding(int... padding)
padding
in class ConvolutionLayer.BaseConvBuilder<Deconvolution3D.Builder>
public void setKernelSize(int... kernelSize)
public void setStride(int... stride)
public void setPadding(int... padding)
public void setDilation(int... dilation)
public Deconvolution3D.Builder dataFormat(Convolution3D.DataFormat dataFormat)
public Deconvolution3D build()
build
in class Layer.Builder<Deconvolution3D.Builder>
Copyright © 2022. All rights reserved.