Concatenation of tensors

mitsuba-renderer / drjit

Dr.Jit — A Just-In-Time-Compiler for Differentiable Rendering

BSD 3-Clause "New" or "Revised" License

593 stars 43 forks source link

Concatenation of tensors #71

Closed maxfrei750 closed 2 years ago

maxfrei750 commented 2 years ago

Is there a way to concatenate multiple TensorXf objects, similar to torch.cat?

Speierers commented 2 years ago

No currently Dr.Jit doesn't provide such a function. Although this shouldn't be to hard to implement using dr.gather and dr.scatter calls. If you give this a try, please submit a PR, I will be happy to assist you to get it pushed onto the master branch.