jittor.distributions
这里是Jittor的随机分布模块的API文档,您可以通过from jittor import distributions
来获取该模块。
-
class jittor.distributions.Categorical(probs=None, logits=None)[源代码]
-
entropy()[源代码]
-
log_prob(x)[源代码]
-
sample(sample_shape=())[源代码]
-
class jittor.distributions.GammaDistribution(concentration, rate)[源代码]
For now only support gamma distribution.
-
cdf(value)[源代码]
-
log_prob(value)[源代码]
-
mean()[源代码]
-
mode()[源代码]
-
sample(shape)[源代码]
-
variance()[源代码]
-
class jittor.distributions.Geometric(p=None, logits=None)[源代码]
-
entropy()[源代码]
-
log_prob(x)[源代码]
-
sample(sample_shape)[源代码]
-
class jittor.distributions.Normal(mu, sigma)[源代码]
-
entropy()[源代码]
-
log_prob(x)[源代码]
-
sample(sample_shape=None)[源代码]
-
class jittor.distributions.OneHotCategorical(probs=None, logits=None)[源代码]
-
entropy()[源代码]
-
log_prob(x)[源代码]
-
sample(sample_shape=[])[源代码]
-
class jittor.distributions.Uniform(low, high)[源代码]
-
entropy()[源代码]
-
log_prob(x)[源代码]
-
sample(sample_shape)[源代码]
-
jittor.distributions.kl_divergence(cur_dist, old_dist)[源代码]
-
jittor.distributions.simple_presum(x)[源代码]