pantheonrl.algos.adap.util.get_categorical

get_categorical(ctx_size, num, use_torch=False)[source]

Samples from categorical distribution