Explanation needed for the shape of my independent variable

Hey guys,
I am struggling at the moment with understanding the shape of my independent variable. Here it is:

torch.Size([64, 3, 224, 224])

Alright, so I understand that the 64 represents the number of batches. 224x224 represents the size of my images.
But what does the 3 represent?

Also bonus question, which dimension is dimension 1? The 64 or the 224?

3 means you have 3 channels color: Red, Green, Blue

for dimension, you mean the index ? 0, 1,2,3 corresponding to 64, 3, 224, 224

1 Like

Seriously??? Thank you so much. I was searching deep through the documentation to figure out where that value was coming from.

If I could like your comment twice, I would.


1 Like