Код Junior Генеративные модели обновлено
Репараметризация и KL
Условие
Для диагональной гауссианы с параметрами mu и logvar напишите vae_step(mu, logvar, eps): верните [z, kl], где z = mu + exp(0.5·logvar)·eps (поэлементно), а kl = -0.5·Σ(1 + logvar - mu² - exp(logvar)) это KL до N(0, I). Всё округлите до 4 знаков.
Примеры
vae_step([0, 0], [0, 0], [1, -1]) // [[1, -1], 0]
vae_step([1], [0], [0]) // [[1], 0.5]
vae_step([0], [1.3862943611198906], [0.5]) // [[1], 0.8069]
Решите в тренажёре
Редактор с автотестами прямо в браузере. Можно решить спокойно, вслух или под таймером, как на собесе.