Проектирование модели: Класс nn.Module

Создание нейросети в PyTorch похоже на сборку конструктора LEGO. В блоке _init_ мы объявляем детали (слои), а в методе forward прописываем инструкцию: как входной тензор должен пройти через эти детали, чтобы превратиться в предсказание. Главное преимущество nn.Module — он автоматически отслеживает все параметры (веса) внутри своих слоев.

Full content is available with a subscription.
Get full access to all courses on the platform for one year with a single payment.
Unlike other platforms that charge per course, here you get everything for one price, and after one year of use there will be no automatic charge for the following year.