Set more precise shape to the attention weights and outputs (#1) a9697cc johnhew ivanzhouyq commited on Aug 14, 2023