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

Moved .all() checks for distributions to _is_all_true (#145029)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/145029
Approved by: https://github.com/Skylion007, https://github.com/zou3519
This commit is contained in:
chilli
2025-01-16 17:04:42 -08:00
committed by PyTorch MergeBot
parent 2bf772d1ba
commit 5e4cf3e6ad
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
torch/distributions/distribution.py
View File

@ -68,7 +68,7 @@ class Distribution:
continue # skip checking lazily-constructed args
value = getattr(self, param)
valid = constraint.check(value)
if not valid.all():
if not torch._is_all_true(valid):
raise ValueError(
f"Expected parameter {param} "
f"({type(value).__name__} of shape {tuple(value.shape)}) "
@ -313,7 +313,7 @@ class Distribution:
return
assert support is not None
valid = support.check(value)
if not valid.all():
if not torch._is_all_true(valid):
raise ValueError(
"Expected value argument "
f"({type(value).__name__} of shape {tuple(value.shape)}) "