bc51dbc0c56f68ed30857755026633f78eef1ae8,spotlight/layers.py,BloomEmbedding,forward,#BloomEmbedding#Any#,171

Before Change

            embedding = self.embeddings(masked_indices[:, :, 0])

            for idx in range(1, len(self._masks)):
                embedding += self.embeddings(masked_indices[:, :, idx])

            embedding /= len(self._masks)

        return embedding

After Change

        masked_indices = Variable(masked_indices)

        embedding = self.embeddings(masked_indices)
        embedding = embedding.sum(1)
        embedding = embedding.view(batch_size, seq_size, -1)

        return embedding

Instances