So my initial input is a dataset of around 800 rows and 64 columns (tokenised texts of length 64).
I’m using a batch size of 64, and my data consists of tokenised texts of length 64, so the I print the size of b_input_ids, I get [64,64].
Any ideas what Pytorch is doing to the batch given it doesn’t have this third dimension like in your example?