mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
[ONNX] Fix doc typo for symbolic_multi_out (#160702)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/160702 Approved by: https://github.com/justinchuby
This commit is contained in:
committed by
PyTorch MergeBot
parent
bbd11c4f23
commit
e299926f72
@ -208,7 +208,7 @@ def symbolic_multi_out(
|
||||
|
||||
# Create a symbolic ONNX operator with the name "CustomOp" in the "custom_domain" domain.
|
||||
# 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",
|
||||
(x,),
|
||||
dict(attr_key="attr_value"),
|
||||
|
Reference in New Issue
Block a user