pytorch_categorical
