Code

Add cumulative sum tensor operation#1722

Code

Closed

allenqm wants to merge 11 commits intotracel-ai:maintracel-ai/burn:mainfrom allenqm:cumulative_sumallenqm/burn:cumulative_sumCopy head branch name to clipboard

+389-2

Commits

Commits on May 3, 2024

• Begin draft of cumulative sum: interfaces

  allenqm

  committed
  b7188aa

  Copy full SHA for b7188aa

Commits on May 7, 2024

• format

  allenqm

  committed
  c51d7be

  Copy full SHA for c51d7be

• add torch implementation

  allenqm

  committed
  729778e

  Copy full SHA for 729778e

• add candle implementation

  allenqm

  committed
  ed7681e

  Copy full SHA for ed7681e

• add ndarray implementation

  allenqm

  committed
  9175f92

  Copy full SHA for 9175f92

• add autodiff implementation

  allenqm

  committed
  3a477a2

  Copy full SHA for 3a477a2

• add tensor tests

  allenqm

  committed
  46cc2c8

  Copy full SHA for 46cc2c8

• remove int overflow test

  allenqm

  committed
  91c27b4

  Copy full SHA for 91c27b4

• add cumsum source to burn-import. tests/ in subsequent commits.

  allenqm

  committed
  fdc5c95

  Copy full SHA for fdc5c95

Commits on May 8, 2024

• add burn import tests

  allenqm

  committed
  8b9b39e

  Copy full SHA for 8b9b39e

• rename cumsum_dim to cumsum

  allenqm

  committed
  dab2905

  Copy full SHA for dab2905