接受领域神经网络

Question

我有一个带有PyTorch的神经网络，如下所示：

class Network(nn.Module):
    def __init__(self):
        super(Network, self).__init__()
        self.fc1 = nn.Linear(784, 200)
        self.fc2 = nn.Linear(200, 150)
        self.fc3 = nn.Linear(150, 100)
        self.fc4 = nn.Linear(100, 10)

我已经在MNIST数据集上对其进行了训练，现在我试图可视化隐藏神经元子集的感受野。对于第一个隐藏层，我这样做是这样的：

neuron = 1
net.load_state_dict(torch.load('saved_model'))

w1 = net.fc1.weight.data

rc1 = w1[neuron,:]
rc1 = rc1.reshape(28,28)
plt.imshow(rc1,)
plt.show()

但是现在我也想在第二层和第三层中可视化一些神经元。但是我应该怎么做呢？