Code

Add scatter operations#255

Code

Merged

CarloLucibello merged 57 commits intoFluxML:masterFluxML/NNlib.jl:masterfrom yuehhua:scatteryuehhua/NNlib.jl:scatterCopy head branch name to clipboardMar 5, 2021

+363

Commits

Commits on Jan 11, 2021

• add scatter operations

  yuehhua

  committed
  a0a5841

• Generalize scatter with op argument and accept AbstractArray

  yuehhua

  committed
  81ff677

• Update src/scatter.jl

  Show description for b2b1b58
  

  yuehhua

  and

  CarloLucibello

  committed
  b2b1b58

• Update src/scatter.jl

  Show description for 2a9f250
  

  yuehhua

  and

  CarloLucibello

  committed
  2a9f250

• Update src/utils.jl

  Show description for 73a03f0
  

  yuehhua

  and

  CarloLucibello

  committed
  73a03f0

• Separate scatter mean doc

  yuehhua

  committed
  a9ffc85

• Drop redundant line of code

  yuehhua

  committed
  b102691

• Update doc and variable names

  yuehhua

  committed
  99a5998

• Fix variable name

  yuehhua

  committed
  6478cbc

• Update docs

  yuehhua

  committed
  aa02132

• Add gather

  yuehhua

  committed
  f063cb9

• Add least_dims

  Show description for 1731b1b
  yuehhua

  committed
  1731b1b

• Support index represented in tuple

  Show description for e50b808
  yuehhua

  committed
  e50b808

• Merge gather! functions

  yuehhua

  committed
  436e38f

• Fix typo

  yuehhua

  committed
  582b141

• Scatter: scalar version

  yuehhua

  committed
  3768de1

• Add dims argument

  Show description for 09ed77a
  yuehhua

  committed
  09ed77a

• Update src/scatter.jl

  Show description for f5f2c93
  

  yuehhua

  and

  chengchingwen

  committed
  f5f2c93

• Update src/gather.jl

  Show description for 0464869
  

  yuehhua

  and

  mcabbott

  committed
  0464869

• Do not need colons

  yuehhua

  committed
  57e3dce

• Update src/scatter.jl

  Show description for a5a6630
  

  yuehhua

  and

  mcabbott

  committed
  a5a6630

• Support gather for scalar and array version

  yuehhua

  committed
  625de76

• Add bound checks for gather

  yuehhua

  committed
  12c526c

• Temporally drop gather_indices

  yuehhua

  committed
  40e993c

• Make dims as kwargs

  yuehhua

  committed
  09c43b8

• Fix bug

  yuehhua

  committed
  7f07448

• Add scatter function

  yuehhua

  committed
  2f9d8c2

• Draft for scatter gradient

  yuehhua

  committed
  24a19ca

Commits on Jan 13, 2021

• change zygote-style to chainrules-style and refactor

  yuehhua

  committed
  6651a3e

• replace nothing with DoesNotExist

  yuehhua

  committed
  1356965

• correct with NO_FIELDS

  yuehhua

  committed
  1f80035

• fix code to be compatible with v1.3

  yuehhua

  committed
  70a6409

• extract ∇scatter_src

  yuehhua

  committed
  2bda28c

Commits on Jan 14, 2021

• extract ∇scatter_dst

  Show description for 8e3ad4a
  yuehhua

  committed
  8e3ad4a

Commits on Jan 26, 2021

• bug fix

  yuehhua

  committed
  882a953

Commits on Feb 18, 2021

• add test_rrule for testing gradient

  yuehhua

  committed
  91d8c30

Commits on Feb 24, 2021

• rewrite gather

  CarloLucibello

  committed
  a577725

• project cleanup

  CarloLucibello

  committed
  cd7c449

• add tests for scatter! and scatter

  yuehhua

  committed
  0453f0c

Commits on Feb 25, 2021

• add dimensional check for output arrays

  yuehhua

  committed
  28bd91f

• move gradient of scatter to another PR

  yuehhua

  committed
  f9c5647

• fix conflict

  yuehhua

  committed
  cdb55fc

• move gather implementation to other PR

  yuehhua

  committed
  cebf473

Commits on Feb 26, 2021

• remove Compat

  yuehhua

  committed
  5dfc49b

Commits on Feb 28, 2021

• remove dims args

  yuehhua

  committed
  3783697

• add @inbounds back

  yuehhua

  committed
  4f11c40

Commits on Mar 1, 2021

• remove restriction of numerical types

  yuehhua

  committed
  010af70

• remove type promotion

  yuehhua

  committed
  66ded23

• remove inbounds and simd annotations

  yuehhua

  committed
  dcc6710

• update error message

  yuehhua

  committed
  8faca3d

• remove @BoundsCheck

  yuehhua

  committed
  e7913b0

• fix

  yuehhua

  committed
  e4f0c17

• replace zeros and ones with more generic way

  yuehhua

  committed
  ec402d9

Commits on Mar 2, 2021

• remove bound checks

  yuehhua

  committed
  c6213c6

• optimize

  yuehhua

  committed
  4d5cbe8

Commits on Mar 3, 2021

• update docs

  yuehhua

  committed
  fc7360d

• remove not used utilities

  yuehhua

  committed
  6542ea9