Solved by ptrblck in post #4 
              
              
                If you want to grab the dimension shapes, you could use: 
size = my_tensor.size()
# or
N, C, H, W = my_tensor.size()

.shape should also work.