public class StackVertex extends BaseGraphVertex
epsilon, graph, inputs, inputVertices, outputVertices, vertexIndex, vertexName
Constructor and Description |
---|
StackVertex(ComputationGraph graph,
java.lang.String name,
int vertexIndex) |
StackVertex(ComputationGraph graph,
java.lang.String name,
int vertexIndex,
VertexIndices[] inputVertices,
VertexIndices[] outputVertices) |
Modifier and Type | Method and Description |
---|---|
Pair<Gradient,org.nd4j.linalg.api.ndarray.INDArray[]> |
doBackward(boolean tbptt)
Do backward pass
|
org.nd4j.linalg.api.ndarray.INDArray |
doForward(boolean training)
Do forward pass using the stored inputs
|
Pair<org.nd4j.linalg.api.ndarray.INDArray,MaskState> |
feedForwardMaskArrays(org.nd4j.linalg.api.ndarray.INDArray[] maskArrays,
MaskState currentMaskState,
int minibatchSize) |
Layer |
getLayer()
Get the Layer (if any).
|
boolean |
hasLayer()
Whether the GraphVertex contains a
Layer object or not |
boolean |
isOutputVertex()
Whether the GraphVertex is an output vertex
|
void |
setBackpropGradientsViewArray(org.nd4j.linalg.api.ndarray.INDArray backpropGradientsViewArray)
|
java.lang.String |
toString() |
canDoBackward, canDoForward, clear, getEpsilon, getInputVertices, getNumInputArrays, getNumOutputConnections, getOutputVertices, getVertexIndex, getVertexName, isInputVertex, setEpsilon, setInput, setInputVertices, setLayerAsFrozen, setOutputVertices
clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait
getInputs, setInputs
public StackVertex(ComputationGraph graph, java.lang.String name, int vertexIndex)
public StackVertex(ComputationGraph graph, java.lang.String name, int vertexIndex, VertexIndices[] inputVertices, VertexIndices[] outputVertices)
public boolean hasLayer()
GraphVertex
Layer
object or notpublic boolean isOutputVertex()
GraphVertex
public Layer getLayer()
GraphVertex
GraphVertex.hasLayer()
== falsepublic org.nd4j.linalg.api.ndarray.INDArray doForward(boolean training)
GraphVertex
training
- if true: forward pass at training time. If false: forward pass at test timepublic Pair<Gradient,org.nd4j.linalg.api.ndarray.INDArray[]> doBackward(boolean tbptt)
GraphVertex
tbptt
- If true: do backprop using truncated BPTTpublic void setBackpropGradientsViewArray(org.nd4j.linalg.api.ndarray.INDArray backpropGradientsViewArray)
GraphVertex
public Pair<org.nd4j.linalg.api.ndarray.INDArray,MaskState> feedForwardMaskArrays(org.nd4j.linalg.api.ndarray.INDArray[] maskArrays, MaskState currentMaskState, int minibatchSize)
public java.lang.String toString()
toString
in class BaseGraphVertex