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

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. Use dataFormat(CNN2DFormat)
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 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 size

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

Use dataFormat(CNN2DFormat)

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

build

public SpaceToDepthLayer build()

Specified by:

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