probly.method.prior_network.torch

Torch Prior Network implementation.

Functions

append_activation_torch

Append exp so the model outputs Dirichlet alpha based on [MG18].