Commit f82920ef authored by Adam Roberts's avatar Adam Roberts Committed by Amit Patankar
Browse files

Add multi-head attention capabilities to AttentionWrapper via the... 

Add multi-head attention capabilities to AttentionWrapper via the specification of multiple AttentionMechanisms.

PiperOrigin-RevId: 162557562
parent 18158c87
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment