Python torch.nn.Linear中w是一个矩阵还是一个向量
时间: 2024-05-31 07:10:11 浏览: 160
浅析PyTorch中nn.Linear的使用
在 PyTorch 中,torch.nn.Linear 中的参数 w 是一个矩阵。具体来说,它是一个大小为 (out_features, in_features) 的二维张量,其中 out_features 表示输出特征的数量,in_features 表示输入特征的数量。该矩阵用于将输入特征转换为输出特征。
阅读全文