oneflow.distributionsΒΆ

Distribution

Distribution is the abstract base class for probability distributions.

Categorical

Creates a categorical distribution parameterized by either probs or logits (but not both).