I think I could use .select() and/or .select() to get the views of the original array that I want, but that doesn’t solve the problem of compiling them together into a batch tensor that can be propagated through a model without copying data. NestedTensor looks like a useful project – hope that continues development.