Thanks for the link. I had a look through the history of TensorDataset as the post came from Feb. 2017, and can’t find any constraints regarding 2dimensional tensors.
This would also mean that no image tensors could be used in TensorDataset which would be a strange design.
Another person commented a week later that tensors with an arbitrary number of dimensions can be used, so I guess it’s just a mistake by the user.