Package | Description |
---|---|
org.deeplearning4j.nn.conf | |
org.deeplearning4j.nn.conf.graph | |
org.deeplearning4j.nn.conf.graph.rnn | |
org.deeplearning4j.nn.transferlearning |
Modifier and Type | Field and Description |
---|---|
protected Map<String,GraphVertex> |
ComputationGraphConfiguration.vertices |
protected Map<String,GraphVertex> |
ComputationGraphConfiguration.GraphBuilder.vertices |
Modifier and Type | Method and Description |
---|---|
ComputationGraphConfiguration.GraphBuilder |
ComputationGraphConfiguration.GraphBuilder.addVertex(String vertexName,
GraphVertex vertex,
String... vertexInputs)
Add a
GraphVertex to the network configuration. |
Modifier and Type | Class and Description |
---|---|
class |
ElementWiseVertex
An ElementWiseVertex is used to combine the activations of two or more layer in an element-wise manner
For example, the activations may be combined by addition, subtraction or multiplication. |
class |
L2NormalizeVertex
L2NormalizeVertex performs L2 normalization on a single input.
|
class |
L2Vertex
L2Vertex calculates the L2 least squares error of two inputs.
|
class |
LayerVertex
LayerVertex is a GraphVertex with a neural network Layer (and, optionally an
InputPreProcessor ) in it |
class |
MergeVertex
A MergeVertex is used to combine the activations of two or more layers/GraphVertex by means of concatenation/merging.
Exactly how this is done depends on the type of input. For 2d (feed forward layer) inputs: MergeVertex([numExamples,layerSize1],[numExamples,layerSize2]) -> [numExamples,layerSize1 + layerSize2] For 3d (time series) inputs: MergeVertex([numExamples,layerSize1,timeSeriesLength],[numExamples,layerSize2,timeSeriesLength]) -> [numExamples,layerSize1 + layerSize2,timeSeriesLength] For 4d (convolutional) inputs: MergeVertex([numExamples,depth1,width,height],[numExamples,depth2,width,height]) -> [numExamples,depth1 + depth2,width,height] |
class |
PreprocessorVertex
PreprocessorVertex is a simple adaptor class that allows a
InputPreProcessor to be used in a ComputationGraph
GraphVertex, without it being associated with a layer. |
class |
ScaleVertex
A ScaleVertex is used to scale the size of activations of a single layer
For example, ResNet activations can be scaled in repeating blocks to keep variance under control. |
class |
StackVertex
StackVertex allows for stacking of inputs so that they may be forwarded through
a network.
|
class |
SubsetVertex
SubsetVertex is used to select a subset of the activations out of another GraphVertex.
For example, a subset of the activations out of a layer. Note that this subset is specifying by means of an interval of the original activations. |
class |
UnstackVertex
UnstackVertex allows for unstacking of inputs so that they may be forwarded through
a network.
|
Modifier and Type | Method and Description |
---|---|
GraphVertex |
LayerVertex.clone() |
GraphVertex |
PreprocessorVertex.clone() |
abstract GraphVertex |
GraphVertex.clone() |
Modifier and Type | Class and Description |
---|---|
class |
DuplicateToTimeSeriesVertex
DuplicateToTimeSeriesVertex is a vertex that goes from 2d activations to a 3d time series activations, by means of
duplication.
|
class |
LastTimeStepVertex
LastTimeStepVertex is used in the context of recurrent neural network activations, to go from 3d (time series)
activations to 2d activations, by extracting out the last time step of activations for each example.
This can be used for example in sequence to sequence architectures, and potentially for sequence classification. |
Modifier and Type | Method and Description |
---|---|
GraphVertex |
LastTimeStepVertex.clone() |
GraphVertex |
DuplicateToTimeSeriesVertex.clone() |
Modifier and Type | Method and Description |
---|---|
TransferLearning.GraphBuilder |
TransferLearning.GraphBuilder.addVertex(String vertexName,
GraphVertex vertex,
String... vertexInputs)
Add a vertex of the given configuration to the computation graph
|
Copyright © 2017. All Rights Reserved.