cb1782b75f7c98f1411bdcb2818f3e2389ff2015,pytext/models/embeddings/pretrained_model_embedding.py,PretrainedModelEmbedding,forward,#PretrainedModelEmbedding#Any#,19

Before Change

                f"Input embedding_dim {embedding.shape[1]} is not a"
                + f" multiple of specified embedding_dim {self.embedding_dim}"
            )
        num_tokens = embedding.shape[1] // self.embedding_dim
        unflattened_embedding = embedding.view(-1, num_tokens, self.embedding_dim)
        return unflattened_embedding

After Change

        // Unflatten embedding Tensor from (batch_size, seq_len * embedding_size)
        // to (batch_size, seq_len, embedding_size).
        num_tokens = embedding_shape[1] // self.embedding_dim
        new_embedding_shape = torch.cat(
            (
                torch.LongTensor([-1]),
                num_tokens.view(1),
                torch.LongTensor([self.embedding_dim]),
            )
        )
        return torch.onnx.operators.reshape_from_tensor_shape(
            embedding, new_embedding_shape
        )

Instances