allow tensor in DiagonalNormal dimension#571

arrjon merged 6 commits intodevbayesflow-org/bayesflow:devfrom

normal_distribution_dimensionbayesflow-org/bayesflow:normal_distribution_dimensionCopy head branch name to clipboard