bed4f2dd50ff54eb7629362c3d03a3b758e0745a,spotlight/layers.py,BloomEmbedding,forward,#BloomEmbedding#Any#,115

Before Change

                              .expand(batch_size, self.num_hash_functions))
        masks = self._mask_tensor.expand(batch_size, self.num_hash_functions)
        compressed_indices = compressed_indices.detach()
        compressed_indices = Variable((compressed_indices.data * masks.data) %
                                      self.compressed_num_embeddings)

        return self.embeddings(compressed_indices).mean(1)

After Change

        embedding /= self.num_hash_functions

        return embedding

Instances