Merge pull request #12825 from lhlmgr/patch-1

Update relaxed_onehot_categorical.py
This commit is contained in:
Yifei Feng 2017-09-05 16:53:16 -07:00 committed by GitHub
commit 12a628a623

View File

@ -150,7 +150,7 @@ class ExpRelaxedOneHotCategorical(distribution.Distribution):
`N - 1` dimensions index into a batch of independent distributions and `N - 1` dimensions index into a batch of independent distributions and
the last dimension represents a vector of probabilities for each the last dimension represents a vector of probabilities for each
class. Only one of `logits` or `probs` should be passed in. class. Only one of `logits` or `probs` should be passed in.
dtype: The type of the event samples (default: int32). dtype: The type of the event samples (default: float32).
validate_args: Python `bool`, default `False`. When `True` distribution validate_args: Python `bool`, default `False`. When `True` distribution
parameters are checked for validity despite possibly degrading runtime parameters are checked for validity despite possibly degrading runtime
performance. When `False` invalid inputs may silently render incorrect performance. When `False` invalid inputs may silently render incorrect
@ -388,7 +388,7 @@ class RelaxedOneHotCategorical(
dimensions index into a batch of independent distributions and the last dimensions index into a batch of independent distributions and the last
dimension represents a vector of probabilities for each class. Only one dimension represents a vector of probabilities for each class. Only one
of `logits` or `probs` should be passed in. of `logits` or `probs` should be passed in.
dtype: The type of the event samples (default: int32). dtype: The type of the event samples (default: float32).
validate_args: Unused in this distribution. validate_args: Unused in this distribution.
allow_nan_stats: Python `bool`, default `True`. If `False`, raise an allow_nan_stats: Python `bool`, default `True`. If `False`, raise an
exception if a statistic (e.g. mean/mode/etc...) is undefined for any exception if a statistic (e.g. mean/mode/etc...) is undefined for any