Ed_Lay的博客

原创 Pytorch中的一些模块

Pytorch中的一些模块 1.nn.Linear() nn.Linear（）用来设置神经网络中的全连接层。全连接层的输入和输出均为二维张量，一般形状为[batch_size, size]。这与卷积层所要求的的输入输出量均为四维张量不同。其用法和形参如下： in_features指的是输入的二维张量的大小，即输入的[batch_size, size]中的size。   out_features指的是输出的二维张量的大小，即输出的二维张量的形状为[batch_size，output_size]，当然，它也代