org.deeplearning4j.nn.graph.vertex

Class BaseWrapperVertex

java.lang.Object

org.deeplearning4j.nn.graph.vertex.BaseWrapperVertex

All Implemented Interfaces:

Serializable, Trainable, GraphVertex

Direct Known Subclasses:

FrozenVertex

public abstract class BaseWrapperVertex
extends Object
implements GraphVertex

A base class for wrapper vertices: i.e., those vertices that have another vertex inside. Use this as the basis of such wrapper vertices, which can selectively override only the vertices that are required.

Author:

Alex Black

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected GraphVertex	underlying

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	BaseWrapperVertex(GraphVertex underlying)

Method Summary

Modifier and Type	Method and Description
boolean	canDoBackward() Whether the GraphVertex can do backward pass.
boolean	canDoForward() Whether the GraphVertex can do forward pass.
void	clear() Clear the internal state (if any) of the GraphVertex.
void	clearVertex() This method clears inpjut for this vertex
Pair<Gradient,INDArray[]>	doBackward(boolean tbptt, LayerWorkspaceMgr workspaceMgr) Do backward pass
INDArray	doForward(boolean training, LayerWorkspaceMgr workspaceMgr) Do forward pass using the stored inputs
Pair<INDArray,MaskState>	feedForwardMaskArrays(INDArray[] maskArrays, MaskState currentMaskState, int minibatchSize)
TrainingConfig	getConfig()
INDArray	getEpsilon() Get the epsilon/error (i.e., dL/dOutput) array previously set for this GraphVertex
INDArray	getGradientsViewArray()
INDArray[]	getInputs() Get the array of inputs previously set for this GraphVertex
VertexIndices[]	getInputVertices() A representation of the vertices that are inputs to this vertex (inputs duing forward pass) Specifically, if inputVertices[X].getVertexIndex() = Y, and inputVertices[X].getVertexEdgeNumber() = Z then the Zth output connection (see GraphVertex.getNumOutputConnections() of vertex Y is the Xth input to this vertex
Layer	getLayer() Get the Layer (if any).
int	getNumInputArrays() Get the number of input arrays.
int	getNumOutputConnections() Get the number of outgoing connections from this GraphVertex.
VertexIndices[]	getOutputVertices() A representation of the vertices that this vertex is connected to (outputs duing forward pass) Specifically, if outputVertices[X].getVertexIndex() = Y, and outputVertices[X].getVertexEdgeNumber() = Z then the Xth output of this vertex is connected to the Zth input of vertex Y
int	getVertexIndex() Get the index of the GraphVertex
String	getVertexName() Get the name/label of the GraphVertex
boolean	hasLayer() Whether the GraphVertex contains a Layer object or not
boolean	isInputVertex() Whether the GraphVertex is an input vertex
boolean	isOutputVertex() Whether the GraphVertex is an output vertex
int	numParams()
INDArray	params()
Map<String,INDArray>	paramTable(boolean backpropOnly) Get the parameter table for the vertex
void	setBackpropGradientsViewArray(INDArray backpropGradientsViewArray) See Model.setBackpropGradientsViewArray(INDArray)
void	setEpsilon(INDArray epsilon) Set the errors (epsilon - aka dL/dActivation) for this GraphVertex
void	setInput(int inputNumber, INDArray input, LayerWorkspaceMgr workspaceMgr) Set the input activations.
void	setInputs(INDArray... inputs) Set all inputs for this GraphVertex
void	setInputVertices(VertexIndices[] inputVertices) Sets the input vertices.
void	setLayerAsFrozen() Only applies to layer vertices.
void	setOutputVertex(boolean outputVertex) Set the GraphVertex to be an output vertex
void	setOutputVertices(VertexIndices[] outputVertices) set the output vertices.

Methods inherited from class java.lang.Object

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

Field Detail

underlying

protected GraphVertex underlying

Constructor Detail

BaseWrapperVertex

protected BaseWrapperVertex(GraphVertex underlying)

Method Detail

getVertexName

public String getVertexName()

Description copied from interface: GraphVertex

Get the name/label of the GraphVertex

Specified by:

getVertexName in interface GraphVertex

getVertexIndex

public int getVertexIndex()

Description copied from interface: GraphVertex

Get the index of the GraphVertex

Specified by:

getVertexIndex in interface GraphVertex

getNumInputArrays

public int getNumInputArrays()

Description copied from interface: GraphVertex

Get the number of input arrays. For example, a Layer may have only one input array, but in general a GraphVertex may have an arbtrary (>=1) number of input arrays (for example, from multiple other layers)

Specified by:

getNumInputArrays in interface GraphVertex

getNumOutputConnections

public int getNumOutputConnections()

Description copied from interface: GraphVertex

Get the number of outgoing connections from this GraphVertex. A GraphVertex may only have a single output (for example, the activations out of a layer), but this output may be used as the input to an arbitrary number of other GraphVertex instances. This method returns the number of GraphVertex instances the output of this GraphVertex is input for.

Specified by:

getNumOutputConnections in interface GraphVertex

getInputVertices

public VertexIndices[] getInputVertices()

Description copied from interface: GraphVertex

A representation of the vertices that are inputs to this vertex (inputs duing forward pass)
Specifically, if inputVertices[X].getVertexIndex() = Y, and inputVertices[X].getVertexEdgeNumber() = Z then the Zth output connection (see GraphVertex.getNumOutputConnections() of vertex Y is the Xth input to this vertex

Specified by:

getInputVertices in interface GraphVertex

setInputVertices

public void setInputVertices(VertexIndices[] inputVertices)

Description copied from interface: GraphVertex

Sets the input vertices.

Specified by:

setInputVertices in interface GraphVertex

See Also:

GraphVertex.getInputVertices()

getOutputVertices

public VertexIndices[] getOutputVertices()

Description copied from interface: GraphVertex

A representation of the vertices that this vertex is connected to (outputs duing forward pass) Specifically, if outputVertices[X].getVertexIndex() = Y, and outputVertices[X].getVertexEdgeNumber() = Z then the Xth output of this vertex is connected to the Zth input of vertex Y

Specified by:

getOutputVertices in interface GraphVertex

setOutputVertices

public void setOutputVertices(VertexIndices[] outputVertices)

Description copied from interface: GraphVertex

set the output vertices.

Specified by:

setOutputVertices in interface GraphVertex

See Also:

GraphVertex.getOutputVertices()

hasLayer

public boolean hasLayer()

Description copied from interface: GraphVertex

Whether the GraphVertex contains a Layer object or not

Specified by:

hasLayer in interface GraphVertex

isInputVertex

public boolean isInputVertex()

Description copied from interface: GraphVertex

Whether the GraphVertex is an input vertex

Specified by:

isInputVertex in interface GraphVertex

isOutputVertex

public boolean isOutputVertex()

Description copied from interface: GraphVertex

Whether the GraphVertex is an output vertex

Specified by:

isOutputVertex in interface GraphVertex

setOutputVertex

public void setOutputVertex(boolean outputVertex)

Description copied from interface: GraphVertex

Set the GraphVertex to be an output vertex

Specified by:

setOutputVertex in interface GraphVertex

getLayer

public Layer getLayer()

Description copied from interface: GraphVertex

Get the Layer (if any). Returns null if GraphVertex.hasLayer() == false

Specified by:

getLayer in interface GraphVertex

setInput

public void setInput(int inputNumber,
                     INDArray input,
                     LayerWorkspaceMgr workspaceMgr)

Description copied from interface: GraphVertex

Set the input activations.

Specified by:

setInput in interface GraphVertex

Parameters:

inputNumber - Must be in range 0 to GraphVertex.getNumInputArrays()-1

input - The input array

setEpsilon

public void setEpsilon(INDArray epsilon)

Description copied from interface: GraphVertex

Set the errors (epsilon - aka dL/dActivation) for this GraphVertex

Specified by:

setEpsilon in interface GraphVertex

clear

public void clear()

Description copied from interface: GraphVertex

Clear the internal state (if any) of the GraphVertex. For example, any stored inputs/errors

Specified by:

clear in interface GraphVertex

canDoForward

public boolean canDoForward()

Description copied from interface: GraphVertex

Whether the GraphVertex can do forward pass. Typically, this is just whether all inputs are set.

Specified by:

canDoForward in interface GraphVertex

canDoBackward

public boolean canDoBackward()

Description copied from interface: GraphVertex

Whether the GraphVertex can do backward pass. Typically, this is just whether all errors/epsilons are set

Specified by:

canDoBackward in interface GraphVertex

doForward

public INDArray doForward(boolean training,
                          LayerWorkspaceMgr workspaceMgr)

Description copied from interface: GraphVertex

Do forward pass using the stored inputs

Specified by:

doForward in interface GraphVertex

Parameters:

training - if true: forward pass at training time. If false: forward pass at test time

Returns:

The output (for example, activations) of the GraphVertex

doBackward

public Pair<Gradient,INDArray[]> doBackward(boolean tbptt,
                                            LayerWorkspaceMgr workspaceMgr)

Description copied from interface: GraphVertex

Do backward pass

Specified by:

doBackward in interface GraphVertex

Parameters:

tbptt - If true: do backprop using truncated BPTT

Returns:

The gradients (may be null), and the errors/epsilons for all inputs to this GraphVertex

getInputs

public INDArray[] getInputs()

Description copied from interface: GraphVertex

Get the array of inputs previously set for this GraphVertex

Specified by:

getInputs in interface GraphVertex

getEpsilon

public INDArray getEpsilon()

Description copied from interface: GraphVertex

Get the epsilon/error (i.e., dL/dOutput) array previously set for this GraphVertex

Specified by:

getEpsilon in interface GraphVertex

setInputs

public void setInputs(INDArray... inputs)

Description copied from interface: GraphVertex

Set all inputs for this GraphVertex

Specified by:

setInputs in interface GraphVertex

See Also:

GraphVertex.setInput(int, INDArray, LayerWorkspaceMgr)

getGradientsViewArray

public INDArray getGradientsViewArray()

Specified by:

getGradientsViewArray in interface Trainable

Returns:

1D gradients view array

setBackpropGradientsViewArray

public void setBackpropGradientsViewArray(INDArray backpropGradientsViewArray)

Description copied from interface: GraphVertex

See Model.setBackpropGradientsViewArray(INDArray)

Specified by:

setBackpropGradientsViewArray in interface GraphVertex

feedForwardMaskArrays

public Pair<INDArray,MaskState> feedForwardMaskArrays(INDArray[] maskArrays,
                                                      MaskState currentMaskState,
                                                      int minibatchSize)

Specified by:

feedForwardMaskArrays in interface GraphVertex

setLayerAsFrozen

public void setLayerAsFrozen()

Description copied from interface: GraphVertex

Only applies to layer vertices. Will throw exceptions on others. If applied to a layer vertex it will treat the parameters of the layer within it as constant. Activations through these will be calculated as they would as test time regardless of training mode

Specified by:

setLayerAsFrozen in interface GraphVertex

clearVertex

public void clearVertex()

Description copied from interface: GraphVertex

This method clears inpjut for this vertex

Specified by:

clearVertex in interface GraphVertex

paramTable

public Map<String,INDArray> paramTable(boolean backpropOnly)

Description copied from interface: GraphVertex

Get the parameter table for the vertex

Specified by:

paramTable in interface Trainable

Specified by:

paramTable in interface GraphVertex

Parameters:

backpropOnly - If true: exclude unsupervised training parameters

Returns:

Parameter table

getConfig

public TrainingConfig getConfig()

Specified by:

getConfig in interface Trainable

Returns:

Training configuration

params

public INDArray params()

Specified by:

params in interface Trainable

Returns:

1d parameter vector

numParams

public int numParams()

Specified by:

numParams in interface Trainable

Returns:

Number of parameters