nestedtensor

nestedtensor copied to clipboard

🚀 Feature

A nested tensor version of the scan operators (e.g. cumsum, cumprod).

Motivation

This will be super useful for volume rendering, which involves cumsum / cumprod operations over values sampled along a ray. With regular Tensors, each ray must have the same number of samples.

Hello @tschmidt23,

Thank you for your feature request! cumsum is already implemented, you can see the test here. I'll add cumprod to the list of work to be done, but there's still currently quite a lot of other work in progress. Do you have a list of specific operators you'd like to unblock a particular usecase or project you're currently working on? We could prioritize that.

Thanks, Christian