НачатьНачать бесплатно

Сеть с линейным слоем

Нейронные сети часто состоят из множества слоёв, однако большинство из них — линейные. Разобраться в работе одного линейного слоя — значит понять их принцип до того, как модель станет сложнее.

Примените линейный слой к входному тензору и посмотрите на результат.

Это упражнение является частью курса

Введение в глубокое обучение с PyTorch

Посмотреть курс

Инструкции к упражнению

  • Создайте слой Linear, который принимает 3 признака на вход и возвращает 2 значения на выходе.
  • Передайте input_tensor через линейный слой.

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

import torch
import torch.nn as nn

input_tensor = torch.tensor([[0.3471, 0.4547, -0.2356]])

# Create a Linear layer
linear_layer = nn.____(
                         in_features=____, 
                         out_features=____
                         )

# Pass input_tensor through the linear layer
output = ____(____)

print(output)
Редактировать и запускать код