python - 在pytorch中的不同nn.Moules中共享参数

Question

我已经得到了您可以在下面看到的模型，但我需要创建两个共享 x2h 和 h2h 的实例。有谁知道怎么做吗？

class RNN(nn.Module):
    def __init__(self, input_size, hidden_size, output_size):
        super(RNN, self).__init__()

        self.hidden_size = hidden_size
        self.x2h = nn.Linear(input_size, hidden_size)
        self.h2h = nn.Linear(hidden_size, hidden_size)
        self.h2o = nn.Linear(hidden_size, output_size)

        #self.softmax = nn.LogSoftmax(dim=1)
        self.softmax = nn.Softmax(dim=1)

    def forward(self, input, hidden):

        hidden1 = self.x2h(input)
        hidden2 = self.h2h(hidden)
        hidden = hidden1 + hidden2
        output = self.h2o(hidden)
        output = self.softmax(output)

        return output, hidden

    def initHidden(self):
        return torch.zeros(1, self.hidden_size)

Answer 1

我认为这是一个Python问题。

在类内部而不是在方法内部声明的变量是类变量或静态变量。

引用: https://radek.io/2011/07/21/static-variables-and-methods-in-python/