線性層網路
神經網路通常包含許多層,但大多數是線性層。先理解單一線性層,能幫助你在加入更多複雜度前掌握其運作方式。
將線性層套用到輸入張量,並觀察輸出。
本練習屬於課程
使用 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)