Home
last modified time | relevance | path

Searched defs:arg_constraints (Results 1 – 25 of 40) sorted by relevance

12

/external/pytorch/torch/distributions/
Drelaxed_bernoulli.py41 arg_constraints = {"probs": constraints.unit_interval, "logits": constraints.real} variable in LogitRelaxedBernoulli
130 arg_constraints = {"probs": constraints.unit_interval, "logits": constraints.real} variable in RelaxedBernoulli
Drelaxed_categorical.py37 arg_constraints = {"probs": constraints.simplex, "logits": constraints.real_vector} variable in ExpRelaxedCategorical
118 arg_constraints = {"probs": constraints.simplex, "logits": constraints.real_vector} variable in RelaxedOneHotCategorical
Dchi2.py24 arg_constraints = {"df": constraints.positive} variable in Chi2
Dlogistic_normal.py34 arg_constraints = {"loc": constraints.real, "scale": constraints.positive} variable in LogisticNormal
Dlkj_cholesky.py60 arg_constraints = {"concentration": constraints.positive} variable in LKJCholesky
Dweibull.py29 arg_constraints = { variable in Weibull
Dpareto.py27 arg_constraints = {"alpha": constraints.positive, "scale": constraints.positive} variable in Pareto
Dfishersnedecor.py31 arg_constraints = {"df1": constraints.positive, "df2": constraints.positive} variable in FisherSnedecor
Dgumbel.py31 arg_constraints = {"loc": constraints.real, "scale": constraints.positive} variable in Gumbel
Dlog_normal.py30 arg_constraints = {"loc": constraints.real, "scale": constraints.positive} variable in LogNormal
Dinverse_gamma.py33 arg_constraints = { variable in InverseGamma
Dkumaraswamy.py40 arg_constraints = { variable in Kumaraswamy
Dpoisson.py32 arg_constraints = {"rate": constraints.nonnegative} variable in Poisson
DstudentT.py32 arg_constraints = { variable in StudentT
Dgeometric.py44 arg_constraints = {"probs": constraints.unit_interval, "logits": constraints.real} variable in Geometric
Dhalf_cauchy.py32 arg_constraints = {"scale": constraints.positive} variable in HalfCauchy
Dhalf_normal.py32 arg_constraints = {"scale": constraints.positive} variable in HalfNormal
Dcauchy.py33 arg_constraints = {"loc": constraints.real, "scale": constraints.positive} variable in Cauchy
Dtransformed_distribution.py49 arg_constraints: Dict[str, constraints.Constraint] = {} variable in TransformedDistribution
Dlaplace.py29 arg_constraints = {"loc": constraints.real, "scale": constraints.positive} variable in Laplace
Duniform.py32 arg_constraints = { variable in Uniform
Dmultinomial.py50 arg_constraints = {"probs": constraints.simplex, "logits": constraints.real_vector} variable in Multinomial
Dbeta.py32 arg_constraints = { variable in Beta
Dnegative_binomial.py31 arg_constraints = { variable in NegativeBinomial
Dindependent.py43 arg_constraints: Dict[str, constraints.Constraint] = {} variable in Independent

12