Torch Discounted Cumsum Versions Save

Fast Discounted Cumulative Sums in PyTorch

v1.1.0

2 years ago
  • Batch support for gamma
  • Gradients support for gamma