硬件开发工资待遇_青岛专业网站制作团队_百度网址收录入口_学生没钱怎么开网店

介绍

在 PyTorch 中，nn.Linear 是一个用于实现全连接层（线性层）的模块。它的作用是对输入数据进行线性变换，公式如下：

$$y=x{W}^T+b$$

其中：

• $x$ 是输入张量
• $W$ 是权重矩阵
• $b$ 是偏置向量（如果 bias=True）

torch.nn.Linear(p_input, p_output,bias)

• p_input: 输入数据的变量个数
• p_output: 输出数据的变量个数
• bias: 是否使用偏置

实例

import torch  
import torch.nn as nn  # 定义一个线性层  
linear = nn.Linear(4, 3, bias=True)  # 查看权重和偏置的形状  
print(linear.weight.shape)  # torch.Size([3, 4]) -> 输出特征数 x 输入特征数  
print(linear.bias.shape)    # torch.Size([3]) -> 输出特征数  # 输入一个张量  
x = torch.rand(2, 4)  # 输入形状为 (batch_size=2, input_features=4)  
output = linear(x)    # 输出形状为 (batch_size=2, output_features=3)  
print(output.shape)   # torch.Size([2, 3])