protected abstract static class SubsamplingLayer.BaseSubsamplingBuilder<T extends SubsamplingLayer.BaseSubsamplingBuilder<T>> extends Layer.Builder<T>
Modifier and Type | Field and Description |
---|---|
protected boolean |
avgPoolIncludePadInDivisor |
protected ConvolutionMode |
convolutionMode
Set the convolution mode for the Convolution layer.
|
protected boolean |
cudnnAllowFallback
When using CuDNN and an error is encountered, should fallback to the non-CuDNN implementatation be allowed?
If set to false, an exception in CuDNN will be propagated back to the user.
|
protected double |
eps |
protected int[] |
kernelSize |
protected int[] |
padding |
protected int |
pnorm |
protected PoolingType |
poolingType |
protected int[] |
stride |
allParamConstraints, biasConstraints, iDropout, layerName, weightConstraints
Modifier | Constructor and Description |
---|---|
protected |
BaseSubsamplingBuilder(int... kernelSize) |
protected |
BaseSubsamplingBuilder(int[] kernelSize,
int[] stride) |
protected |
BaseSubsamplingBuilder(int[] kernelSize,
int[] stride,
int[] padding) |
protected |
BaseSubsamplingBuilder(PoolingType poolingType) |
protected |
BaseSubsamplingBuilder(PoolingType poolingType,
int[] kernelSize) |
protected |
BaseSubsamplingBuilder(PoolingType poolingType,
int[] kernelSize,
int[] stride,
int[] padding) |
protected |
BaseSubsamplingBuilder(SubsamplingLayer.PoolingType poolingType) |
protected |
BaseSubsamplingBuilder(SubsamplingLayer.PoolingType poolingType,
int[] kernelSize) |
protected |
BaseSubsamplingBuilder(SubsamplingLayer.PoolingType poolingType,
int[] kernelSize,
int[] stride) |
protected |
BaseSubsamplingBuilder(SubsamplingLayer.PoolingType poolingType,
int[] kernelSize,
int[] stride,
int[] padding) |
Modifier and Type | Method and Description |
---|---|
protected abstract boolean |
allowCausal() |
T |
avgPoolIncludePadInDivisor(boolean avgPoolIncludePadInDivisor)
When doing average pooling, should the padding values be included in the divisor or not?
Not applicable for max and p-norm pooling. Users should not usually set this - instead, leave it as the default (false). |
T |
convolutionMode(ConvolutionMode convolutionMode)
Set the convolution mode for the Convolution layer.
|
T |
cudnnAllowFallback(boolean allowFallback)
Deprecated.
|
T |
eps(double eps) |
T |
helperAllowFallback(boolean allowFallback)
When using CuDNN or MKLDNN and an error is encountered, should fallback to the non-helper implementation be allowed?
If set to false, an exception in the helper will be propagated back to the user.
|
T |
pnorm(int pnorm) |
T |
poolingType(PoolingType poolingType) |
T |
poolingType(SubsamplingLayer.PoolingType poolingType) |
void |
setConvolutionMode(ConvolutionMode convolutionMode) |
void |
setEps(double eps) |
void |
setPnorm(int pnorm) |
build, constrainAllParameters, constrainBias, constrainWeights, dropOut, dropOut, name
protected PoolingType poolingType
protected int[] kernelSize
protected int[] stride
protected int[] padding
protected ConvolutionMode convolutionMode
ConvolutionMode
for more details
Convolution mode for layerprotected int pnorm
protected double eps
protected boolean cudnnAllowFallback
protected boolean avgPoolIncludePadInDivisor
protected BaseSubsamplingBuilder(SubsamplingLayer.PoolingType poolingType, int[] kernelSize, int[] stride)
protected BaseSubsamplingBuilder(SubsamplingLayer.PoolingType poolingType, int[] kernelSize)
protected BaseSubsamplingBuilder(SubsamplingLayer.PoolingType poolingType, int[] kernelSize, int[] stride, int[] padding)
protected BaseSubsamplingBuilder(PoolingType poolingType, int[] kernelSize)
protected BaseSubsamplingBuilder(PoolingType poolingType, int[] kernelSize, int[] stride, int[] padding)
protected BaseSubsamplingBuilder(int[] kernelSize, int[] stride, int[] padding)
protected BaseSubsamplingBuilder(int[] kernelSize, int[] stride)
protected BaseSubsamplingBuilder(int... kernelSize)
protected BaseSubsamplingBuilder(SubsamplingLayer.PoolingType poolingType)
protected BaseSubsamplingBuilder(PoolingType poolingType)
public void setPnorm(int pnorm)
public void setEps(double eps)
protected abstract boolean allowCausal()
public void setConvolutionMode(ConvolutionMode convolutionMode)
public T convolutionMode(ConvolutionMode convolutionMode)
ConvolutionMode
for more detailsconvolutionMode
- Convolution mode for layerpublic T poolingType(SubsamplingLayer.PoolingType poolingType)
public T poolingType(PoolingType poolingType)
public T pnorm(int pnorm)
public T eps(double eps)
@Deprecated public T cudnnAllowFallback(boolean allowFallback)
helperAllowFallback(boolean)
allowFallback
- Whether fallback to non-CuDNN implementation should be usedpublic T helperAllowFallback(boolean allowFallback)
allowFallback
- Whether fallback to non-CuDNN implementation should be usedpublic T avgPoolIncludePadInDivisor(boolean avgPoolIncludePadInDivisor)
[A, P] [B, P]Where A and B are actual values, and P is padding (0).
avgPoolIncludePadInDivisor
- Whether the divisor should include or exclude padding for average poolingCopyright © 2020. All rights reserved.