pytorch中多维数组某一维度接全连接

在PyTorch中，可以使用`torch.nn.Linear`模块来实现全连接层。假设我们有一个名为`x`的多维张量，它的形状为`(batch_size, dim1, dim2, ..., dimn)`，现在我们想要将第二维度`dim1`接一个全连接层，输出维度为`out_dim`。可以按以下方式实现：

import torch.nn as nn

class MyModel(nn.Module):
    def __init__(self, in_dim, out_dim):
        super(MyModel, self).__init__()
        self.fc = nn.Linear(in_dim, out_dim)

    def forward(self, x):
        # 将第二维度展开
        x = x.view(x.size(0), -1, x.size(-1))
        # 经过全连接层
        x = self.fc(x)
        # 将第二维度恢复原来的形状
        x = x.view(x.size(0), -1, x.size(-1), x.size(-1))
        return x

在`forward`函数中，我们首先使用`view`函数将第二维度展开，然后经过全连接层，最后再使用`view`函数将第二维度恢复原来的形状。