Searched defs:enumerate_support (Results 1 – 6 of 6) sorted by relevance
| /external/pytorch/torch/distributions/ | ||
| D | independent.py | 117 def enumerate_support(self, expand=True): member in Independent |
| D | categorical.py | 151 def enumerate_support(self, expand=True): member in Categorical |
| D | binomial.py | 155 def enumerate_support(self, expand=True): member in Binomial |
| D | bernoulli.py | 120 def enumerate_support(self, expand=True): member in Bernoulli |
| D | one_hot_categorical.py | 110 def enumerate_support(self, expand=True): member in OneHotCategorical |
| D | distribution.py | 218 def enumerate_support(self, expand: bool = True) -> torch.Tensor: member in Distribution |