ML Katas

by: newest upvotes saves

Sparse Updates with `scatter_add_`

this month medium (<30 mins) | scatter_add sparse

In graph neural networks and other sparse data applications, you often need to update a tensor based on sparse indices. Your exercise is to implement a function that takes a tensor of `values`, a...

1