Extend ImageFolder in pytorch to return a tensor of a different shape?

2

Is this a double post from here or are these use cases somehow different?