Class ZeroPaddingLayer.Builder
- java.lang.Object
-
- org.deeplearning4j.nn.conf.layers.Layer.Builder<ZeroPaddingLayer.Builder>
-
- org.deeplearning4j.nn.conf.layers.ZeroPaddingLayer.Builder
-
- Enclosing class:
- ZeroPaddingLayer
public static class ZeroPaddingLayer.Builder extends Layer.Builder<ZeroPaddingLayer.Builder>
-
-
Field Summary
-
Fields inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder
allParamConstraints, biasConstraints, iDropout, layerName, weightConstraints
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description ZeroPaddingLayer
build()
ZeroPaddingLayer.Builder
dataFormat(CNN2DFormat format)
Set the data format for the CNN activations - NCHW (channels first) or NHWC (channels last).void
setPadding(int... padding)
-
Methods inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder
constrainAllParameters, constrainBias, constrainWeights, dropOut, dropOut, name
-
-
-
-
Constructor Detail
-
Builder
public Builder(int padHeight, int padWidth)
- Parameters:
padHeight
- Padding for both the top and bottompadWidth
- Padding for both the left and right
-
Builder
public Builder(int padTop, int padBottom, int padLeft, int padRight)
- Parameters:
padTop
- Top padding valuepadBottom
- Bottom padding valuepadLeft
- Left padding valuepadRight
- Right padding value
-
Builder
public Builder(int[] padding)
- Parameters:
padding
- Must be a length 1 array with values [paddingAll], a length 2 array with values [padTopBottom, padLeftRight], or a length 4 array with values [padTop, padBottom, padLeft, padRight]
-
-
Method Detail
-
dataFormat
public ZeroPaddingLayer.Builder dataFormat(CNN2DFormat format)
Set the data format for the CNN activations - NCHW (channels first) or NHWC (channels last). SeeCNN2DFormat
for more details.
Default: NCHW- Parameters:
format
- Format for activations (in and out)
-
setPadding
public void setPadding(int... padding)
- Parameters:
padding
- Padding value for top, bottom, left, and right. Must be length 4 array
-
build
public ZeroPaddingLayer build()
- Specified by:
build
in classLayer.Builder<ZeroPaddingLayer.Builder>
-
-