Richard/pt 2.10 tensor bugfix

[rsyue]

What does this PR do? Please describe:
Resolved tensor casting issues introduced with PyTorch 2.10 changes, as indicated in #D87484341

This brings the update to OSS.

Does your PR introduce any breaking changes? If yes, please list them:
None known

Check list:

• Was the content of this PR discussed and approved via a GitHub issue? (no need for typos or documentation improvements)
• Did you read the contributor guideline?
• Did you make sure that your PR does only one thing instead of bundling different changes together?
• Did you make sure to update the documentation with your changes? (if necessary)
• Did you write any new necessary tests?
• Did you verify new and existing tests pass locally with your changes?
• Did you update the CHANGELOG? (no need for typos, documentation, or minor internal changes)

[cbalioglu]

Thanks for the PR! Have you verified that the code compiles successfully with PyTorch nightlies?

[rsyue]

Thanks for the PR! Have you verified that the code compiles successfully with PyTorch nightlies?

Everything builds fine with the nightly 2.10 build

[cbalioglu]

LGTM!