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

Add __all__ to various submodules in torch.fx, distributions, distributed, package (#80367)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/80367
Approved by: https://github.com/albanD
This commit is contained in:
anjali411
2022-06-27 18:44:11 +00:00
committed by PyTorch MergeBot
parent 02730c1e79
commit 3bcc19b29a
58 changed files with 70 additions and 455 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
torch/distributions/binomial.py
View File

@ -3,6 +3,7 @@ from torch.distributions import constraints
from torch.distributions.distribution import Distribution
from torch.distributions.utils import broadcast_all, probs_to_logits, lazy_property, logits_to_probs
__all__ = ['Binomial']
def _clamp_by_zero(x):
# works like clamp(x, min=0) but has grad at 0 is 0.5