All Katas - ML Katas

by: newest upvotes saves

Flatten Leading Dimensions

this year easy (<10 mins) | einops reshaping

### Description Given a tensor with multiple leading dimensions, flatten them into a single dimension. For example, transform a tensor of shape `(D1, D2, D3, D4)` into `(D1*D2, D3, D4)`. ###...
Unflatten a Dimension

this year easy (<10 mins) | einops reshaping

### Description This is the inverse of flattening. Given a tensor where the first dimension is a product of two other dimensions, unflatten it. For example, transform a tensor of shape `(D1*D2,...