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.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)