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.
|