添加链接描述
全连接层,相当于tf中的Dense(),用法是nn.Linear(in_features=,out_features=),放在第一层时需要指定输入的维度,即in_features,
out_features是这层网络的神经元个数。
传入Linear层的实际是二维([batch_size,size]),但实际搭建网络层时不用指明batch_size,输入时的维度一定要包含batch_size。
pytorch学习笔记——nn.Linear()
最新推荐文章于 2024-04-04 00:30:00 发布