org.deeplearning4j.nn.conf.layers

Class SpaceToBatchLayer.Builder<T extends SpaceToBatchLayer.Builder<T>>

java.lang.Object

org.deeplearning4j.nn.conf.layers.Layer.Builder<T>

org.deeplearning4j.nn.conf.layers.SpaceToBatchLayer.Builder<T>

Enclosing class:

SpaceToBatchLayer

public static class SpaceToBatchLayer.Builder<T extends SpaceToBatchLayer.Builder<T>>
extends Layer.Builder<T>

Field Summary

Fields

Modifier and Type	Field and Description
protected int[]	blocks
protected int[][]	padding

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

allParamConstraints, biasConstraints, iDropout, layerName, weightConstraints

Constructor Summary

Constructors

Constructor and Description
Builder(int[] blocks)
Builder(int[] blocks, int[][] padding)

Method Summary

Modifier and Type	Method and Description
T	blocks(int[] blocks)
SpaceToBatchLayer	build()
T	name(String layerName) Layer name assigns layer string name.
T	padding(int[][] padding)

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

blocks

protected int[] blocks

padding

protected int[][] padding

Constructor Detail

Builder

public Builder(int[] blocks)

Parameters:

blocks - Block size for SpaceToBatch layer. Should be a length 2 array for the height and width dimensions

Builder

public Builder(int[] blocks,
               int[][] padding)

Parameters:

blocks - Block size for SpaceToBatch layer. Should be a length 2 array for the height and width dimensions

padding - Padding - should be a 2d array, with format [[padTop, padBottom], [padLeft, padRight]]

Method Detail

blocks

public T blocks(int[] blocks)

Parameters:

blocks - Block size for SpaceToBatch layer. Should be a length 2 array for the height and width dimensions

padding

public T padding(int[][] padding)

Parameters:

padding - Padding - should be a 2d array, with format [[padTop, padBottom], [padLeft, padRight]]

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 SpaceToBatchLayer.Builder<T>>

build

public SpaceToBatchLayer build()

Specified by:

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