frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add missing docstring for sym_ite (#154201)

Browse Source 
`sym_ite` is listed in [the reference page](https://docs.pytorch.org/docs/stable/torch.html) and has no document.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/154201
Approved by: https://github.com/Skylion007
This commit is contained in:
Yuki Kobayashi
2025-05-26 15:59:19 +00:00
committed by PyTorch MergeBot
parent 02445ec8f0
commit f55f2f42a7
2 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
torch/__init__.py
View File

@ -978,6 +978,7 @@ __all__.append("sym_sqrt")
def sym_ite(b, t, f):
"""SymInt-aware utility for ternary operator (``t if b else f``.)"""
if overrides.has_torch_function((b, t, f)):
return overrides.handle_torch_function(sym_ite, (b, t, f), b, t, f)
assert isinstance(b, (SymBool, builtins.bool)) and type(t) == type(f)