Support 3D attention mask (#5887)

Commit

5 years ago

Support 3D attention mask (#5887) Support 3D attention mask with shape (batch_size, sequence_length, all_sequence_length)