Concatenating batch-wise traced paths

[naperman]

I want to call scene.trace_paths() batch by batch in a loop and them accumulate(concatenate) the traced paths after every iteration of the loop. The traced paths is an object with several tensors as properties and with the positions of the batch_size dimension differing between tensors in the object(see screenshot below). Is there an easy way to concatenate along batch_size for all properties of traced paths?

The idea is to precompute traced_paths for a large number of UE's sampled on a scene, but by breaking up the compute into runs where traced paths are computed batch by batch. The final combined traced paths object is fed into a train function and reused for calibration of the ray tracer.

[naperman]

@jhoydis Just wanted bring this to your attention because such a paths concat function will be very useful for computing traced paths in batches for a scene with a large number of UEs sampled from the coverage map. I believe paths object is immutable.