Skip to content

Add argument to skip the output reshaping in the attention functions. #859

Add argument to skip the output reshaping in the attention functions.

Add argument to skip the output reshaping in the attention functions. #859