Code

Add axes to indexed and gather#1356

Code

Merged

josevalim merged 4 commits intomainelixir-nx/nx:mainfrom jv-indexed-gather-accesselixir-nx/nx:jv-indexed-gather-accessCopy head branch name to clipboardNov 8, 2023

+390-181

Commits

Commits on Nov 7, 2023

• WIP

  josevalim

  committed
  6ca41df

  Copy full SHA for 6ca41df

• Almost all done

  josevalim

  committed
  3229312

  Copy full SHA for 3229312

• Improve tests

  josevalim

  committed
  3eb44d1

  Copy full SHA for 3eb44d1

Commits on Nov 8, 2023

• feat: support axes in gather/indexed_op

  polvalente

  committed
  7f5ec97

  Copy full SHA for 7f5ec97