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