org.deeplearning4j.arbiter.layers

Class GlobalPoolingLayerSpace.Builder

java.lang.Object

org.deeplearning4j.arbiter.layers.LayerSpace.Builder<GlobalPoolingLayerSpace.Builder>

org.deeplearning4j.arbiter.layers.GlobalPoolingLayerSpace.Builder

Enclosing class:

GlobalPoolingLayerSpace

public static class GlobalPoolingLayerSpace.Builder
extends LayerSpace.Builder<GlobalPoolingLayerSpace.Builder>

Field Summary

Fields

Modifier and Type	Field and Description
protected ParameterSpace<Boolean>	collapseDimensions
protected ParameterSpace<Integer>	pNorm
protected ParameterSpace<int[]>	poolingDimensions
protected ParameterSpace<org.deeplearning4j.nn.conf.layers.PoolingType>	poolingType

Fields inherited from class org.deeplearning4j.arbiter.layers.LayerSpace.Builder

dropOut

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
GlobalPoolingLayerSpace	build()
GlobalPoolingLayerSpace.Builder	collapseDimensions(boolean collapseDimensions)
GlobalPoolingLayerSpace.Builder	collapseDimensions(ParameterSpace<Boolean> collapseDimensions)
GlobalPoolingLayerSpace.Builder	pNorm(int pNorm)
GlobalPoolingLayerSpace.Builder	pNorm(ParameterSpace<Integer> pNorm)
GlobalPoolingLayerSpace.Builder	poolingDimensions(int... poolingDimensions)
GlobalPoolingLayerSpace.Builder	poolingDimensions(ParameterSpace<int[]> poolingDimensions)
GlobalPoolingLayerSpace.Builder	poolingType(ParameterSpace<org.deeplearning4j.nn.conf.layers.PoolingType> poolingType)
GlobalPoolingLayerSpace.Builder	poolingType(org.deeplearning4j.nn.conf.layers.PoolingType poolingType)

Methods inherited from class org.deeplearning4j.arbiter.layers.LayerSpace.Builder

dropOut, dropOut, iDropOut

Methods inherited from class java.lang.Object

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

Field Detail

poolingDimensions

protected ParameterSpace<int[]> poolingDimensions

collapseDimensions

protected ParameterSpace<Boolean> collapseDimensions

poolingType

protected ParameterSpace<org.deeplearning4j.nn.conf.layers.PoolingType> poolingType

pNorm

protected ParameterSpace<Integer> pNorm

Constructor Detail

Builder

public Builder()

Method Detail

poolingDimensions

public GlobalPoolingLayerSpace.Builder poolingDimensions(int... poolingDimensions)

poolingDimensions

public GlobalPoolingLayerSpace.Builder poolingDimensions(ParameterSpace<int[]> poolingDimensions)

collapseDimensions

public GlobalPoolingLayerSpace.Builder collapseDimensions(boolean collapseDimensions)

collapseDimensions

public GlobalPoolingLayerSpace.Builder collapseDimensions(ParameterSpace<Boolean> collapseDimensions)

poolingType

public GlobalPoolingLayerSpace.Builder poolingType(org.deeplearning4j.nn.conf.layers.PoolingType poolingType)

poolingType

public GlobalPoolingLayerSpace.Builder poolingType(ParameterSpace<org.deeplearning4j.nn.conf.layers.PoolingType> poolingType)

pNorm

public GlobalPoolingLayerSpace.Builder pNorm(int pNorm)

pNorm

public GlobalPoolingLayerSpace.Builder pNorm(ParameterSpace<Integer> pNorm)

build

public GlobalPoolingLayerSpace build()

Specified by:

build in class LayerSpace.Builder<GlobalPoolingLayerSpace.Builder>