ML Katas

by: newest upvotes saves

Replicating `torch.nn.Embedding` with `gather`

this month medium (<30 mins) | embeddings NLP gather

The `torch.nn.Embedding` layer is fundamental in many deep learning models, especially in NLP. Your task is to replicate its forward pass functionality using `torch.gather`. You'll create a...

1