pytorch学习笔记——nn.Linear()

添加链接描述

全连接层，相当于tf中的Dense()，用法是nn.Linear(in_features=,out_features=)，放在第一层时需要指定输入的维度，即in_features，

out_features是这层网络的神经元个数。

传入Linear层的实际是二维（[batch_size,size]),但实际搭建网络层时不用指明batch_size，输入时的维度一定要包含batch_size。