Uses of Class
org.deeplearning4j.nn.conf.layers.LearnedSelfAttentionLayer.Builder
-
Packages that use LearnedSelfAttentionLayer.Builder Package Description org.deeplearning4j.nn.conf.layers -
-
Uses of LearnedSelfAttentionLayer.Builder in org.deeplearning4j.nn.conf.layers
Methods in org.deeplearning4j.nn.conf.layers that return LearnedSelfAttentionLayer.Builder Modifier and Type Method Description LearnedSelfAttentionLayer.BuilderLearnedSelfAttentionLayer.Builder. headSize(int headSize)Size of attention headsLearnedSelfAttentionLayer.BuilderLearnedSelfAttentionLayer.Builder. nHeads(int nHeads)Number of Attention HeadsLearnedSelfAttentionLayer.BuilderLearnedSelfAttentionLayer.Builder. nIn(int nIn)LearnedSelfAttentionLayer.BuilderLearnedSelfAttentionLayer.Builder. nOut(int nOut)LearnedSelfAttentionLayer.BuilderLearnedSelfAttentionLayer.Builder. nQueries(int nQueries)Number of queries to learnLearnedSelfAttentionLayer.BuilderLearnedSelfAttentionLayer.Builder. projectInput(boolean projectInput)Project input before applying attention or not.Constructors in org.deeplearning4j.nn.conf.layers with parameters of type LearnedSelfAttentionLayer.Builder Constructor Description LearnedSelfAttentionLayer(LearnedSelfAttentionLayer.Builder builder)
-