开始使用免费开始使用

线性层网络

神经网络通常包含许多层,但其中大多数是线性层。先理解单个线性层,有助于在增加复杂性之前掌握其工作原理。

将线性层应用到一个输入张量,并观察输出。

本练习是课程的一部分

使用 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)
编辑并运行代码