tutorials icon indicating copy to clipboard operation
tutorials copied to clipboard
verified publisher icon pytorch

Fix NestedTensor deprecation warning in SDPA tutorial

Open dorbittonn opened this issue 1 month ago • 3 comments

Add layout=torch.jagged parameter to nested_tensor call to suppress the prototype API warning and use the recommended jagged layout.

Checklist

  • [ ] Only one issue is addressed in this pull request
  • [ ] Labels from the issue that this PR is fixing are added to this pull request
  • [ ] No unnecessary issues are included into this pull request.

dorbittonn avatar Nov 28 '25 19:11 dorbittonn