NestedTensor 29 is a package to work with tensors of different sizes and lengths. It should be released soon, so I guess you might be using the nightly binary?
Anyway, I’m not sure if .shape should return this error or some struct containing different lengths.
CC @cpuhrsch who is the main author of this package.