org.deeplearning4j.nn.graph.vertex

Class BaseGraphVertex

java.lang.Object

org.deeplearning4j.nn.graph.vertex.BaseGraphVertex

All Implemented Interfaces:

java.io.Serializable, GraphVertex

Direct Known Subclasses:

DuplicateToTimeSeriesVertex, ElementWiseVertex, InputVertex, L2NormalizeVertex, L2Vertex, LastTimeStepVertex, LayerVertex, MergeVertex, PreprocessorVertex, ScaleVertex, StackVertex, SubsetVertex, UnstackVertex

public abstract class BaseGraphVertex
extends java.lang.Object
implements GraphVertex

BaseGraphVertex defines a set of common functionality for GraphVertex instances.

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected org.nd4j.linalg.api.ndarray.INDArray	epsilon
protected ComputationGraph	graph
protected org.nd4j.linalg.api.ndarray.INDArray[]	inputs
protected VertexIndices[]	inputVertices A representation of the vertices that are inputs to this vertex (inputs during forward pass) Specifically, if inputVertices[X].getVertexIndex() = Y, and inputVertices[X].getVertexEdgeNumber() = Z then the Zth output of vertex Y is the Xth input to this vertex
protected VertexIndices[]	outputVertices 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 output of this vertex (there is only one output) is connected to the Zth input of vertex Y
protected int	vertexIndex The index of this vertex
protected java.lang.String	vertexName

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	BaseGraphVertex(ComputationGraph graph, java.lang.String name, int vertexIndex, VertexIndices[] inputVertices, VertexIndices[] outputVertices)

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.
org.nd4j.linalg.api.ndarray.INDArray	getEpsilon() Get the epsilon/error (i.e., dL/dOutput) array 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 of vertex Y is the Xth input to this vertex
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
java.lang.String	getVertexName() Get the name/label of the GraphVertex
boolean	isInputVertex() Whether the GraphVertex is an input vertex
void	setEpsilon(org.nd4j.linalg.api.ndarray.INDArray epsilon) Set the errors (epsilon - aka dL/dActivation) for this GraphVertex
void	setInput(int inputNumber, org.nd4j.linalg.api.ndarray.INDArray input) Set the input activations.
void	setInputVertices(VertexIndices[] inputVertices) Sets the input vertices.
void	setLayerAsFrozen() Only applies to layer vertices.
void	setOutputVertices(VertexIndices[] outputVertices) set the output vertices.
abstract java.lang.String	toString()

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.deeplearning4j.nn.graph.vertex.GraphVertex

doBackward, doForward, feedForwardMaskArrays, getInputs, getLayer, hasLayer, isOutputVertex, setBackpropGradientsViewArray, setInputs

Field Detail

graph

protected ComputationGraph graph

vertexName

protected java.lang.String vertexName

vertexIndex

protected int vertexIndex

The index of this vertex

inputVertices

protected VertexIndices[] inputVertices

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

outputVertices

protected VertexIndices[] outputVertices

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 output of this vertex (there is only one output) is connected to the Zth input of vertex Y

inputs

protected org.nd4j.linalg.api.ndarray.INDArray[] inputs

epsilon

protected org.nd4j.linalg.api.ndarray.INDArray epsilon

Constructor Detail

BaseGraphVertex

protected BaseGraphVertex(ComputationGraph graph,
                          java.lang.String name,
                          int vertexIndex,
                          VertexIndices[] inputVertices,
                          VertexIndices[] outputVertices)

Method Detail

getVertexName

public java.lang.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()

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 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()

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()

isInputVertex

public boolean isInputVertex()

Description copied from interface: GraphVertex

Whether the GraphVertex is an input vertex

Specified by:

isInputVertex in interface GraphVertex

setInput

public void setInput(int inputNumber,
                     org.nd4j.linalg.api.ndarray.INDArray input)

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(org.nd4j.linalg.api.ndarray.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

getEpsilon

public org.nd4j.linalg.api.ndarray.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

toString

public abstract java.lang.String toString()

Overrides:

toString in class java.lang.Object

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