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)

Builder

public Builder(int blockSize,
               SpaceToDepthLayer.DataFormat dataFormat)

Method Detail

blocks

public T blocks(int blockSize)

dataFormat

public T dataFormat(SpaceToDepthLayer.DataFormat dataFormat)

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>>