Class Upsampling2D.Builder
- java.lang.Object
-
- org.deeplearning4j.nn.conf.layers.Layer.Builder<T>
-
- org.deeplearning4j.nn.conf.layers.BaseUpsamplingLayer.UpsamplingBuilder<Upsampling2D.Builder>
-
- org.deeplearning4j.nn.conf.layers.Upsampling2D.Builder
-
- Enclosing class:
- Upsampling2D
public static class Upsampling2D.Builder extends BaseUpsamplingLayer.UpsamplingBuilder<Upsampling2D.Builder>
-
-
Field Summary
Fields Modifier and Type Field Description protected CNN2DFormat
format
-
Fields inherited from class org.deeplearning4j.nn.conf.layers.BaseUpsamplingLayer.UpsamplingBuilder
size
-
Fields inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder
allParamConstraints, biasConstraints, iDropout, layerName, weightConstraints
-
-
Constructor Summary
Constructors Constructor Description Builder(int size)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description Upsampling2D
build()
Upsampling2D.Builder
dataFormat(CNN2DFormat format)
Set the data format for the CNN activations - NCHW (channels first) or NHWC (channels last).void
setSize(int... size)
Upsampling2D.Builder
size(int size)
Upsampling size int, used for both height and widthUpsampling2D.Builder
size(int[] size)
Upsampling size array-
Methods inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder
constrainAllParameters, constrainBias, constrainWeights, dropOut, dropOut, name
-
-
-
-
Field Detail
-
format
protected CNN2DFormat format
-
-
Method Detail
-
dataFormat
public Upsampling2D.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)
-
size
public Upsampling2D.Builder size(int size)
Upsampling size int, used for both height and width- Parameters:
size
- upsampling size in height and width dimensions
-
size
public Upsampling2D.Builder size(int[] size)
Upsampling size array- Parameters:
size
- upsampling size in height and width dimensions
-
build
public Upsampling2D build()
- Specified by:
build
in classLayer.Builder<Upsampling2D.Builder>
-
setSize
public void setSize(int... size)
-
-