Update aten_index_put implementation

[gramalingam]

Update aten_index_put implementation in torchlib.

[codecov]

Codecov Report

❌ Patch coverage is 93.02326% with 3 lines in your changes missing coverage. Please review.
✅ Project coverage is 70.06%. Comparing base (c1bfdfc) to head (4cea291).
⚠️ Report is 3 commits behind head on main.
✅ All tests successful. No failed tests found.

Files with missing lines	Patch %	Lines
onnxscript/function_libs/torch_lib/ops/core.py	93.02%	2 Missing and 1 partial ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #2712      +/-   ##
==========================================
+ Coverage   70.04%   70.06%   +0.02%     
==========================================
  Files         226      226              
  Lines       27177    27262      +85     
  Branches     2734     2753      +19     
==========================================
+ Hits        19036    19101      +65     
- Misses       7197     7210      +13     
- Partials      944      951       +7     

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