当初在学习nn.Linear时了解到的博客都是关于一维变换的，比如输入3通道，输出6通道；又比如得到（3，4，4）的特征图，需要进行拉平为（48，）的向量，然后通过nn.Linear（48，10）得到10个输出（...

浏览 72 次 标签： torch.nn.Linear的维度变换过程详解（有图有公式有代码）