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)[源代码]