pepy.tech

torch-discounted-cumsum

1.1.0
8.41k

Fast discounted cumulative sums in PyTorch