Package | Description |
---|---|
org.deeplearning4j.nn.conf.graph | |
org.deeplearning4j.nn.conf.layers.samediff | |
org.deeplearning4j.nn.layers.samediff | |
org.deeplearning4j.nn.params |
Modifier and Type | Class and Description |
---|---|
class |
AttentionVertex |
Modifier and Type | Class and Description |
---|---|
class |
SameDiffLambdaVertex |
Modifier and Type | Field and Description |
---|---|
protected SameDiffVertex |
SameDiffGraphVertex.config |
Constructor and Description |
---|
SameDiffGraphVertex(SameDiffVertex config,
ComputationGraph graph,
String name,
int vertexIndex,
INDArray paramsView,
boolean initParams,
DataType dataType) |
Modifier and Type | Method and Description |
---|---|
Map<String,INDArray> |
SameDiffParamInitializer.subsetAndReshape(List<String> params,
Map<String,long[]> paramShapes,
INDArray view,
AbstractSameDiffLayer sdl,
SameDiffVertex sdv) |
Copyright © 2022. All rights reserved.