1 write to word_embeddings
Microsoft.ML.TorchSharp (1)
Roberta\Modules\Embeddings.cs (1)
25word_embeddings = torch.nn.Embedding(numEmbeddings, embeddingSize, padding_idx: 1);
2 references to word_embeddings
Microsoft.ML.TorchSharp (2)
Roberta\Modules\Embeddings.cs (2)
37var tokenEmbedding = word_embeddings.forward(tokens); 52word_embeddings.Dispose();