Package | Description |
---|---|
org.deeplearning4j.nn.conf.layers |
Modifier and Type | Method and Description |
---|---|
LearnedSelfAttentionLayer.Builder |
LearnedSelfAttentionLayer.Builder.headSize(int headSize)
Size of attention heads
|
LearnedSelfAttentionLayer.Builder |
LearnedSelfAttentionLayer.Builder.nHeads(int nHeads)
Number of Attention Heads
|
LearnedSelfAttentionLayer.Builder |
LearnedSelfAttentionLayer.Builder.nIn(int nIn) |
LearnedSelfAttentionLayer.Builder |
LearnedSelfAttentionLayer.Builder.nOut(int nOut) |
LearnedSelfAttentionLayer.Builder |
LearnedSelfAttentionLayer.Builder.nQueries(int nQueries)
Number of queries to learn
|
LearnedSelfAttentionLayer.Builder |
LearnedSelfAttentionLayer.Builder.projectInput(boolean projectInput)
Project input before applying attention or not.
|
Constructor and Description |
---|
LearnedSelfAttentionLayer(LearnedSelfAttentionLayer.Builder builder) |
Copyright © 2020. All rights reserved.