I was always under impression that arrays should be converted to tenors in __getitem__. It’s shown in the tutorial: https://pytorch.org/tutorials/beginner/data_loading_tutorial.html 37
That would make some sense since some kinds of data cannot be gathered into array until the collate_fn, e.g. text data. Why would they make the memory grow out of bound? I thought that CPU tensors are just wrappers for ndarray.