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

[distributions] Rename .params to .arg_constraints, fix logic (#5989)

Browse Source
This commit is contained in:
Fritz Obermeyer
2018-03-25 06:24:32 -07:00
committed by Adam Paszke
parent 03a6952ac9
commit b2da9fd220
29 changed files with 52 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torch/distributions/binomial.py
View File

@ -31,7 +31,7 @@ class Binomial(Distribution):
probs (Tensor): Event probabilities
logits (Tensor): Event log-odds
"""
params = {'probs': constraints.unit_interval}
arg_constraints = {'probs': constraints.unit_interval}
has_enumerate_support = True
def __init__(self, total_count=1, probs=None, logits=None, validate_args=None):