eagerpy icon indicating copy to clipboard operation
eagerpy copied to clipboard

Missing support for ep.nonzero() and ep.flatnonzero()

Open zimmerrol opened this issue 4 years ago • 1 comments

I think it would be good to have an eagerpy version of the nonzero() and flatnonzero() functions. I guess these can be imitated with a combination of arange and indexing, but it would be more convenient to have a simple drop-in replacement for these two common functions.

zimmerrol avatar Apr 15 '21 09:04 zimmerrol

Sure, if we can implement this for all frameworks, we could add it.

jonasrauber avatar Apr 19 '21 08:04 jonasrauber