開始使用免費開始

線性層網路

神經網路通常包含許多層,但大多數是線性層。先理解單一線性層,能幫助你在加入更多複雜度前掌握其運作方式。

將線性層套用到輸入張量,並觀察輸出。

本練習屬於課程

使用 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)
編輯並執行程式碼