fix fused_head_and_loss_fn bug

[danleifeng]

Before submitting

• Lint code. If there are lint issues, please format the code first.

# Install and register `pre-commit` in the project folder
pip install pre-commit && pre-commit install

# Process previous code files separately
pre-commit run --file XXXX.py

• Add test cases into tests folder. If there are codecov issues, please add tests cases first.

PR types

PR changes

Description

修复开启use_fused_head_and_loss_fn后转置参数传递错误的问题 from #2648

[paddle-bot]

Thanks for your contribution!

[codecov-commenter]

Codecov Report

✅ All modified and coverable lines are covered by tests.
⚠️ Please upload report for BASE (develop@d17af14). Learn more about missing BASE report.

Additional details and impacted files

@@            Coverage Diff             @@
##             develop    #2774   +/-   ##
==========================================
  Coverage           ?   29.03%           
==========================================
  Files              ?      335           
  Lines              ?    56120           
  Branches           ?        0           
==========================================
  Hits               ?    16292           
  Misses             ?    39828           
  Partials           ?        0           

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

[lugimzzz]

LGTM

[lugimzzz]

LGTM