Categorical

Contents

Categorical#

class stable_ssl.data.Categorical(values: list | Tensor, probabilities: list | Tensor)[source]#

Bases: Module

Categorical distribution for sampling discrete values with given probabilities.