nn.Linear的输出
时间: 2024-06-01 19:04:55 浏览: 9
nn.Linear是PyTorch中的一个模块,用于实现线性变换。它将输入张量与权重矩阵相乘,并加上偏置向量,得到输出张量。nn.Linear的输出是一个具有指定形状的张量。
具体来说,假设输入张量的形状为(batch_size, input_size),其中batch_size表示批次大小,input_size表示输入特征的维度。nn.Linear的权重矩阵的形状为(output_size, input_size),其中output_size表示输出特征的维度。偏置向量的形状为(output_size,),即一个一维向量。
那么nn.Linear的输出张量的形状为(batch_size, output_size),即每个样本对应一个输出特征向量。
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)