индекс находится вне диапазона в .. \ 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))

Ожидается решениеэта ошибка.

Спасибо.