Uses of Class
org.deeplearning4j.nn.conf.graph.AttentionVertex.Builder

Packages that use AttentionVertex.Builder

Package	Description
org.deeplearning4j.nn.conf.graph

Uses of AttentionVertex.Builder in org.deeplearning4j.nn.conf.graph

Methods in org.deeplearning4j.nn.conf.graph that return AttentionVertex.Builder

Modifier and Type	Method and Description
AttentionVertex.Builder	AttentionVertex.Builder.headSize(long headSize) Size of Attention Heads
AttentionVertex.Builder	AttentionVertex.Builder.nHeads(int nHeads) Number of Attention Heads
AttentionVertex.Builder	AttentionVertex.Builder.nInKeys(long nInKeys) Size of Keys
AttentionVertex.Builder	AttentionVertex.Builder.nInQueries(long nInQueries) Size of Queries
AttentionVertex.Builder	AttentionVertex.Builder.nInValues(long nInValues) Size of Values
AttentionVertex.Builder	AttentionVertex.Builder.nOut(long nOut) Output Size
AttentionVertex.Builder	AttentionVertex.Builder.projectInput(boolean projectInput) Toggle to enable / disable projection of inputs (key, values, queries).
AttentionVertex.Builder	AttentionVertex.Builder.weightInit(WeightInit weightInit) Weight initialization scheme

Constructors in org.deeplearning4j.nn.conf.graph with parameters of type AttentionVertex.Builder

Constructor and Description
AttentionVertex(AttentionVertex.Builder builder)