在scaled_dot_product_attention函数中，改为如果输入是3D，输出也是3D by Qin-sx · Pull Request #73804 · PaddlePaddle/Paddle
added 6 commits
June 14, 2025 22:17
modified: python/paddle/nn/functional/flash_attention.py modified: test/legacy_test/test_scaled_dot_product_attention.py
modified: test/legacy_test/test_scaled_dot_product_attention.py
modified: test/legacy_test/test_scaled_dot_product_attention.py
modified: python/paddle/nn/functional/flash_attention.py modified: test/legacy_test/test_scaled_dot_product_attention.py
modified: python/paddle/nn/functional/flash_attention.py
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode characters