public class UnstackVertex extends GraphVertex
Modifier and Type | Field and Description |
---|---|
protected int |
from |
protected int |
stackSize |
Constructor and Description |
---|
UnstackVertex(int from,
int stackSize) |
Modifier and Type | Method and Description |
---|---|
UnstackVertex |
clone() |
boolean |
equals(java.lang.Object o) |
InputType |
getOutputType(int layerIndex,
InputType... vertexInputs)
Determine the type of output for this GraphVertex, given the specified inputs.
|
int |
hashCode() |
GraphVertex |
instantiate(ComputationGraph graph,
java.lang.String name,
int idx,
org.nd4j.linalg.api.ndarray.INDArray paramsView,
boolean initializeParams)
Create a
GraphVertex instance, for the given computation graph,
given the configuration instance. |
int |
numParams(boolean backprop) |
public UnstackVertex(int from, int stackSize)
from
- The first column index of the stacked inputs.stackSize
- The total number of stacked inputs. An interval is automatically
calculated according to the size of the first dimension.public GraphVertex instantiate(ComputationGraph graph, java.lang.String name, int idx, org.nd4j.linalg.api.ndarray.INDArray paramsView, boolean initializeParams)
GraphVertex
GraphVertex
instance, for the given computation graph,
given the configuration instance.instantiate
in class GraphVertex
graph
- The computation graph that this GraphVertex is to be part ofname
- The name of the GraphVertex objectidx
- The index of the GraphVertexparamsView
- A view of the full parameters arrayinitializeParams
- If true: initialize the parameters. If false: make no change to the values in the paramsView arraypublic UnstackVertex clone()
clone
in class GraphVertex
public boolean equals(java.lang.Object o)
equals
in class GraphVertex
public int numParams(boolean backprop)
numParams
in class GraphVertex
public int hashCode()
hashCode
in class GraphVertex
public InputType getOutputType(int layerIndex, InputType... vertexInputs) throws InvalidInputTypeException
GraphVertex
getOutputType
in class GraphVertex
layerIndex
- The index of the layer (if appropriate/necessary).vertexInputs
- The inputs to this vertexInvalidInputTypeException
- If the input type is invalid for this type of GraphVertex