как пиро AutoDiagonalNormal определяет размерность скрытого пространства
Пайро-учебник Байесовской регрессии дает пример, выходное скрытое измерение локального и масштаба нормального распределения равно 5. Но когда я изменяю сеть, чтобы иметь 2 линейных слоя, скрытое пространство автоматически изменяется на 10 измерений.
Интересно, как автогид pyro-ppl определяет количество измерений в скрытом пространстве, и может ли пользователь установить это измерение?