Class SpaceToDepthLayer.Builder<T extends SpaceToDepthLayer.Builder<T>>
- java.lang.Object
-
- org.deeplearning4j.nn.conf.layers.Layer.Builder<T>
-
- org.deeplearning4j.nn.conf.layers.SpaceToDepthLayer.Builder<T>
-
- Enclosing class:
- SpaceToDepthLayer
public static class SpaceToDepthLayer.Builder<T extends SpaceToDepthLayer.Builder<T>> extends Layer.Builder<T>
-
-
Field Summary
Fields Modifier and Type Field Description protected int
blockSize
protected CNN2DFormat
dataFormat
Data format for input activations.-
Fields inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder
allParamConstraints, biasConstraints, iDropout, layerName, weightConstraints
-
-
Constructor Summary
Constructors Constructor Description Builder(int blockSize)
Builder(int blockSize, CNN2DFormat dataFormat)
Builder(int blockSize, SpaceToDepthLayer.DataFormat dataFormat)
Deprecated.
-
Method Summary
All Methods Instance Methods Concrete Methods Deprecated Methods Modifier and Type Method Description T
blocks(int blockSize)
SpaceToDepthLayer
build()
T
dataFormat(CNN2DFormat dataFormat)
Set the data format for the CNN activations - NCHW (channels first) or NHWC (channels last).T
dataFormat(SpaceToDepthLayer.DataFormat dataFormat)
Deprecated.T
name(String layerName)
Layer name assigns layer string name.-
Methods inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder
constrainAllParameters, constrainBias, constrainWeights, dropOut, dropOut
-
-
-
-
Field Detail
-
blockSize
protected int blockSize
-
dataFormat
protected CNN2DFormat dataFormat
Data format for input activations. Note DL4J uses NCHW in most cases
-
-
Constructor Detail
-
Builder
public Builder(int blockSize)
- Parameters:
blockSize
- Block size
-
Builder
@Deprecated public Builder(int blockSize, SpaceToDepthLayer.DataFormat dataFormat)
Deprecated.- Parameters:
blockSize
- Block sizedataFormat
- Data format for input activations. Note DL4J uses NCHW in most cases
-
Builder
public Builder(int blockSize, CNN2DFormat dataFormat)
-
-
Method Detail
-
blocks
public T blocks(int blockSize)
- Parameters:
blockSize
- Block size
-
dataFormat
@Deprecated public T dataFormat(SpaceToDepthLayer.DataFormat dataFormat)
Deprecated.- Parameters:
dataFormat
- Data format for input activations. Note DL4J uses NCHW in most cases
-
dataFormat
public T dataFormat(CNN2DFormat dataFormat)
Set the data format for the CNN activations - NCHW (channels first) or NHWC (channels last). SeeCNN2DFormat
for more details.
Default: NCHW- Parameters:
dataFormat
- Format for activations (in and out)
-
name
public T name(String layerName)
Description copied from class:Layer.Builder
Layer name assigns layer string name. Allows easier differentiation between layers.- Overrides:
name
in classLayer.Builder<T extends SpaceToDepthLayer.Builder<T>>
-
build
public SpaceToDepthLayer build()
- Specified by:
build
in classLayer.Builder<T extends SpaceToDepthLayer.Builder<T>>
-
-