public class SubsetVertex extends BaseGraphVertex
epsilon, graph, inputs, inputVertices, outputVertices, vertexIndex, vertexName
Constructor and Description |
---|
SubsetVertex(ComputationGraph graph,
java.lang.String name,
int vertexIndex,
int from,
int to) |
SubsetVertex(ComputationGraph graph,
java.lang.String name,
int vertexIndex,
VertexIndices[] inputVertices,
VertexIndices[] outputVertices,
int from,
int to) |
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 SubsetVertex(ComputationGraph graph, java.lang.String name, int vertexIndex, int from, int to)
public SubsetVertex(ComputationGraph graph, java.lang.String name, int vertexIndex, VertexIndices[] inputVertices, VertexIndices[] outputVertices, int from, int to)
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 java.lang.String toString()
toString
in class BaseGraphVertex
public void setBackpropGradientsViewArray(org.nd4j.linalg.api.ndarray.INDArray backpropGradientsViewArray)
GraphVertex