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

Мережа з лінійним шаром

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

Застосуйте лінійний шар до вхідного тензора й подивіться на результат.

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

Вступ до Deep Learning з 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)
Редагувати та запускати код