Словарь модулей в PyTorch
Создайте модуль с двумя линейными
фрагментами под ключами
left и right:
первый 2 на 3,
второй 3 на 1.
В прямом проходе пропустите
вход через оба по порядку
и выведите форму результата.
Опишите блок, где под ключом
core лежит линейный слой
4 на 4.
Выведите общее число обучаемых
чисел у корневого модуля.
Соберите сеть с тремя линейными
частями в словаре по ключам
a, b и c
размеров 1 на 2,
2 на 2 и 2
на 1. Выведите число
отдельных тензоров параметров
у всей модели.
Создайте модуль с парой
линейных связей 3 на 2
под ключами enc
и dec. Обработайте
вектор [[1.0, 0.0, 1.0]]
только через фрагмент
enc и выведите форму
полученного тензора.
