Model Details. In the distributional layers, standard deviations are obtained by applying a softplus operator to the outputs of the network. The reparametrization trick [1]} is used to propagate gradients through the sampling from the posterior.
