NNlib.jl icon indicating copy to clipboard operation
NNlib.jl copied to clipboard

Published 20 hours ago • verified publisher icon FluxML

add dims keyword to gather

Open CarloLucibello opened this issue 2 years ago • 0 comments

Extend with the dims keyword similarly to https://pytorch.org/docs/stable/generated/torch.gather.html#torch-gather TODO

  • [ ] docs
  • [ ] more tests
  • [ ] gradient
  • [ ] non-mutating version

CarloLucibello avatar Dec 26 '22 13:12 CarloLucibello