org.deeplearning4j.nn.conf.layers

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 and Description
protected int	blockSize
protected SpaceToDepthLayer.DataFormat	dataFormat

Fields inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder

allParamConstraints, biasConstraints, iDropout, layerName, weightConstraints

Constructor Summary

Constructors

Constructor and Description
Builder(int blockSize)
Builder(int blockSize, SpaceToDepthLayer.DataFormat dataFormat)

Method Summary

Modifier and Type	Method and Description
T	blocks(int blockSize)
SpaceToDepthLayer	build()
T	dataFormat(SpaceToDepthLayer.DataFormat dataFormat)
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

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

blockSize

protected int blockSize

dataFormat

protected SpaceToDepthLayer.DataFormat dataFormat

Constructor Detail

Builder

public Builder(int blockSize)

Parameters:

blockSize - Block size

Builder

public Builder(int blockSize,
               SpaceToDepthLayer.DataFormat dataFormat)

Parameters:

blockSize - Block size

dataFormat - Data format for input activations. Note DL4J uses NCHW in most cases

Method Detail

blocks

public T blocks(int blockSize)

Parameters:

blockSize - Block size

dataFormat

public T dataFormat(SpaceToDepthLayer.DataFormat dataFormat)

Parameters:

dataFormat - Data format for input activations. Note DL4J uses NCHW in most cases

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 class Layer.Builder<T extends SpaceToDepthLayer.Builder<T>>

build

public SpaceToDepthLayer build()

Specified by:

build in class Layer.Builder<T extends SpaceToDepthLayer.Builder<T>>