Nested Tensor Guide to nested tensors: https://pytorch.org/tutorials/prototype/nestedtensor.html#why-nested-tensor