nn.linear()函数

  • nn.Linear():用于设置网络中的全连接层,需要注意的是全连接层的输入与输出都是二维张量

  • 一般形状为[batch_size, size],不同于卷积层要求输入输出是四维张量。其用法与形参说明如下:

 

posted @ 2022-03-08 11:53  Tomorrow1126  阅读(1044)  评论(0编辑  收藏  举报