Skip to content

Commit

Permalink
comment
Browse files Browse the repository at this point in the history
  • Loading branch information
Dobiasd committed Dec 16, 2023
1 parent 84c492e commit 3bfbf07
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions include/fdeep/layers/multi_head_attention_layer.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,7 @@ class multi_head_attention_layer : public layer
//const tensor& value = input[1];
//const tensor& key = input.size() > 2 ? input[2] : value;
// https://github.com/keras-team/keras/blob/v2.14.0/keras/layers/attention/multi_head_attention.py
// https://gist.github.com/sevagh/b71d253a347a9b59c026580625452fc5
return input;
}
std::size_t num_heads_;
Expand Down

0 comments on commit 3bfbf07

Please sign in to comment.