시작하기무료로 시작하기

선형 계층 네트워크

Neural networks에는 여러 계층이 있지만, 그중 상당수는 선형 계층입니다. 먼저 단일 선형 계층을 이해하면, 복잡도를 더하기 전에 동작 원리를 파악하는 데 도움이 됩니다.

입력 텐서에 선형 계층을 적용하고 출력을 확인해 보세요.

이 연습은 강의의 일부입니다

PyTorch로 배우는 딥러닝 입문

강의 보기

연습 안내

  • 입력으로 3개의 feature를 받고 2개의 출력을 반환하는 Linear 계층을 생성하세요.
  • 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)
코드 편집 및 실행