线性层网络
神经网络通常包含许多层,但其中大多数是线性层。先理解单个线性层,有助于在增加复杂性之前掌握其工作原理。
将线性层应用到一个输入张量,并观察输出。
本练习是课程的一部分
使用 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)