PyTorch的torch.nn.linear()用于设置网络当中的全连接层,输入和输出都是二维张量,而二维卷积层要求输入和输出是四维张量。
in_features指的是输入的二维张量的大小,即输入的[batch_size, size]中的size。
out_features指的是输出的二维张量的大小,即输出的二维张量的形状为[batch_size,output_size],当然,它也代表了该全连接层的神经元个数。
从输入输出的张量的shape角度来理解,相当于一个输入为[batch_size, in_features]的张量变换成了[batch_size, out_features]的输出张量。
参考博文https://blog.csdn.net/qq_42079689/article/details/102873766