定义一个简单的线性网络,连接输入和输出的全连接层
时间: 2024-06-09 10:10:07 浏览: 83
一个简单的网络框架
PyTorch中定义一个简单的线性网络,连接输入和输出的全连接层可以使用nn.Linear()函数。这个函数的参数包括输入特征的维度(in_features)和输出特征的维度(out_features),同时还可以选择是否包含偏置项(bias)。以下是定义一个简单的线性网络的代码示例:
```
import torch
import torch.nn as nn
# 定义输入特征维度
input_features = 10
# 定义输出特征维度
output_features = 5
# 定义全连接层
linear_layer = nn.Linear(input_features, output_features)
# 打印全连接层
print(linear_layer)
```
阅读全文