ПочатиПочніть безкоштовно

Ваша перша нейронна мережа

Час реалізувати невелику нейронну мережу з двох лінійних шарів, розташованих послідовно.

Ця вправа є частиною курсу

Вступ до Deep Learning з PyTorch

Переглянути курс

Інструкції до вправи

  • Додайте контейнер для послідовного складання шарів.

Інтерактивна практична вправа

Спробуйте виконати цю вправу, доповнивши цей зразок коду.

import torch
import torch.nn as nn

input_tensor = torch.Tensor([[2, 3, 6, 7, 9, 3, 2, 1]])

# Create a container for stacking linear layers
model = nn.____(nn.Linear(8, 4),
                nn.Linear(4, 1)
                )

output = model(input_tensor)
print(output)
Редагувати та запускати код