关于pytorch中nn.linear输入维度的问题

因为最近看了一下attention博客，寻思挺简单，想动手写一个，才动手就发现有个问题，对于Q,K,V矩阵，看似只是个简单的前馈网络结构就能实现，但是在多数问题中，输入的都是词向量，也就是说，输入数据的结构是(batch_size，seq_size，embeding_size)，而nn.linear一般接受的输入格式是（输入神经元个数，输出神经元个数）。

经过查验，这里我发现nn.linear其实可以接受任意维度输入，但只改变最后一维的输出，比如我输入为（20，10, 30），这里20为一个batch大小，10为单词量，30为词向量维度，设输出为18个神经元

则nn.linear（30,18）

输出的维度为（20,10,18）