Add default values in docs
@@ -233,8 +233,9 @@ class Attention(BaseDenseAttention):
causal: Boolean. Set to `True` for decoder self-attention. Adds a mask such
that position `i` cannot attend to positions `j > i`. This prevents the
flow of information from the future towards the past.
+ Defaults to `False`.
dropout: Float between 0 and 1. Fraction of the units to drop for the
- attention scores.
+ attention scores. Defaults to 0.0.
Call Args:
@@ -372,8 +373,9 @@ class AdditiveAttention(BaseDenseAttention):
causal: Boolean. Set to `True` for decoder self-attention. Adds a mask such
that position `i` cannot attend to positions `j > i`. This prevents the
flow of information from the future towards the past.
+ Defaults to `False`.
dropout: Float between 0 and 1. Fraction of the units to drop for the
- attention scores.
+ attention scores. Defaults to 0.0.
Call Args:
