to then compute the distributions (during training...
# datascience
c
to then compute the distributions (during training)