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
Implements Dot Product Attention using the given inputs.
|
Modifier and Type | Class and Description |
---|---|
class |
SameDiffLambdaVertex
SameDiffLambdaVertex is defined to be used as the base class for implementing lambda vertices using SameDiff
Lambda vertices are vertices without parameters - and as a result, have a much simpler API - users need only extend SameDiffLambdaVertex and implement a single method to define their vertex |
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 © 2020. All rights reserved.