New adapter transforms: subsetting arrays

[paul-buerkner]

Sometimes, the simulators may return larger arrays than we want to use in our networks, so it would be great to have some subsetting adapter transforms.

We need:

• Subsetting within an axis (taking only some elements) while keeping the number of axes the same. This is essentially the np.take functionality so we might want to call this transform take. In contrast to np.take, I would make axis a mandatory arguments or default it to the last axis. Example: adapter.take("x", 1:3, axis = -1)
• Subsetting using a random set of indices (of user-specified size) within an axis. We might call this subsample. Internally it would call take after sampling the indices. Example: adapter.subsample("x", size = 3, axis = -1)
• Removing an axis of length one. Following numpy, I would call this transform squeeze: Example: adapter.squeeze("x", axis = 1)

[CorentinChab]

Hello @paul-buerkner , can I be assigned to this issue ?

[marvinschmitt]

Hello @paul-buerkner , can I be assigned to this issue ?

I have assigned you, thanks for volunteering to work on this issue!

[CorentinChab]

Thanks ! @stefanradev93 told me there was a Slack channel. Can I join so we can agree on the test specification I am writing, before developping actual implementation ? It's my first time doing something like this so I might have some questions 😓 Hope not be a burden haha

[marvinschmitt]

Thanks ! @stefanradev93 told me there was a Slack channel. Can I join so we can agree on the test specification I am writing, before developping actual implementation ? It's my first time doing something like this so I might have some questions 😓 Hope not be a burden haha

No worries at all, all contributions are greatly appreciated and you're more than welcome to ask all the questions you have at any point :)

Could you please share your email address to mail.marvinschmitt@gmail.com, then I'll send you a Slack invite.

[eodole]

Hello @CorentinChab @marvinschmitt I had talked to @paul-buerkner about working on this issue but didnt actually get assigned to it. I already had some work done on it. @CorentinChab would you want to collaborate on this?

[CorentinChab]

Yup definitely @eodole !