Is there a torch implementation of distribution layers, so it is possible to predict uncertainty in Neural Networks? Something similar to Tensorflow Probability or Yarin Gal's concrete dropout implementation.
Hi @fg.cordoni ,
We have support for a few distributions, and they can be use within
nn_modules, here's an example: