OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[ONNX] Fix doc typo for symbolic_multi_out (#160702)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/160702
Approved by: https://github.com/justinchuby
This commit is contained in:
Shiva Kaul 2025-08-15 14:34:42 +00:00 committed by PyTorch MergeBot
parent bbd11c4f23
commit e299926f72
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
torch/onnx/ops/__init__.py
View File

@ -208,7 +208,7 @@ def symbolic_multi_out(
# Create a symbolic ONNX operator with the name "CustomOp" in the "custom_domain" domain. # Create a symbolic ONNX operator with the name "CustomOp" in the "custom_domain" domain.
# The output tensors will have the specified dtypes and shapes # The output tensors will have the specified dtypes and shapes
(out1, out2) = torch.onnx.ops.symbolic( (out1, out2) = torch.onnx.ops.symbolic_multi_out(
"custom_domain::CustomOp", "custom_domain::CustomOp",
(x,), (x,),
dict(attr_key="attr_value"), dict(attr_key="attr_value"),