distr_categorical function

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

torch packageRead PDF manual
  • Maintainer: Daniel Falbel
  • License: MIT + file LICENSE
  • Last published: 2024-05-21

Useful links