Cumulative sum with decay factor

6

Check out a package I made for this purpose: GitHub - toshas/torch-discounted-cumsum: Fast Discounted Cumulative Sums in PyTorch