Mạng với tầng tuyến tính
Neural networks thường có nhiều tầng, nhưng phần lớn là tầng tuyến tính. Hiểu rõ một tầng tuyến tính sẽ giúp bạn nắm được cách chúng hoạt động trước khi thêm độ phức tạp.
Hãy áp dụng một tầng tuyến tính lên một tensor đầu vào và quan sát đầu ra.
Bài tập này là một phần của khóa học
Nhập môn Deep Learning với PyTorch
Hướng dẫn bài tập
- Tạo một tầng
Linearnhận 3 đặc trưng làm đầu vào và trả về 2 đầu ra. - Truyền
input_tensorqua tầng tuyến tính.
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
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)