索引超出范围.. \ aten \ src \ TH / generic / THTensorEvenMoreMath.cpp：193

Question

我正在尝试构建联结树vae，我为这个错误而感到震惊。我的嵌入大小与词汇量大小相同，另一个参数是隐藏的大小长度。我要去哪里错了？请帮助某人。

我已经阅读了许多博客，并尝试解决此错误，但是我做不到。我知道嵌入存在一些问题，但是我不知道如何解决？

def __init__(self, vocab, hidden_size, latent_size, depthT, depthG):
        super(JTNNVAE, self).__init__()
        self.vocab = vocab
        #print(int(vocab.size()))
        self.hidden_size = int(hidden_size)
        self.latent_size = latent_size = latent_size / 2 #Tree and Mol has two vectors
        self.latent_size=int(self.latent_size)
        self.jtnn = JTNNEncoder(hidden_size, depthT, nn.Embedding(int(vocab.size()), int(hidden_size)))
        self.decoder = JTNNDecoder(vocab, hidden_size, latent_size, nn.Embedding(int(vocab.size()), int(hidden_size)))

        self.jtmpn = JTMPN(hidden_size, depthG)
        self.mpn = MPN(hidden_size, depthG)

        self.A_assm = nn.Linear(int(latent_size), int(hidden_size), bias=False)
        self.assm_loss = nn.CrossEntropyLoss(size_average=False)

        self.T_mean = nn.Linear(int(hidden_size), int(latent_size))
        self.T_var = nn.Linear(int(hidden_size), int(latent_size))
        self.G_mean = nn.Linear(int(hidden_size), int(latent_size))
        self.G_var = nn.Linear(int(hidden_size), int(latent_size))

有望解决此错误。

谢谢。