In torch.nn.Embedding(num_embeddings, embedding_dim, …), the depth of the table (i.e. num_embeddings value) must be defined with a constant value in a given NN model. What if, in this model, num_embeddings itself is a variable? How to cope with it?
Any insight in this regard will be highly appreciated.