org.deeplearning4j.arbiter.layers

Class GlobalPoolingLayerSpace

java.lang.Object

org.deeplearning4j.arbiter.optimize.api.AbstractParameterSpace<L>

org.deeplearning4j.arbiter.layers.LayerSpace<org.deeplearning4j.nn.conf.layers.GlobalPoolingLayer>

org.deeplearning4j.arbiter.layers.GlobalPoolingLayerSpace

All Implemented Interfaces:

ParameterSpace<org.deeplearning4j.nn.conf.layers.GlobalPoolingLayer>

public class GlobalPoolingLayerSpace
extends LayerSpace<org.deeplearning4j.nn.conf.layers.GlobalPoolingLayer>

Layer space for a GlobalPoolingLayer

Author:

Alex Black

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	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

dropOut

Method Summary

Modifier and Type	Method and Description
org.deeplearning4j.nn.conf.layers.GlobalPoolingLayer	getValue(double[] parameterValues)
boolean	isLeaf()
int	numParameters()
void	setIndices(int... indices)

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

collectLeaves, setLayerOptionsBuilder, toString, toString

Methods inherited from class org.deeplearning4j.arbiter.optimize.api.AbstractParameterSpace

getNestedSpaces

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, 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

Method Detail

getValue

public org.deeplearning4j.nn.conf.layers.GlobalPoolingLayer getValue(double[] parameterValues)

numParameters

public int numParameters()

Specified by:

numParameters in interface ParameterSpace<org.deeplearning4j.nn.conf.layers.GlobalPoolingLayer>

Overrides:

numParameters in class LayerSpace<org.deeplearning4j.nn.conf.layers.GlobalPoolingLayer>

isLeaf

public boolean isLeaf()

Specified by:

isLeaf in interface ParameterSpace<org.deeplearning4j.nn.conf.layers.GlobalPoolingLayer>

Overrides:

isLeaf in class LayerSpace<org.deeplearning4j.nn.conf.layers.GlobalPoolingLayer>

setIndices

public void setIndices(int... indices)

Specified by:

setIndices in interface ParameterSpace<org.deeplearning4j.nn.conf.layers.GlobalPoolingLayer>

Overrides:

setIndices in class LayerSpace<org.deeplearning4j.nn.conf.layers.GlobalPoolingLayer>