@tom
torch.Size([4, 272, 352])  my tensor shape is this…  0,1 is one row in actual image , 2,3 is second row. But when i do reshape to get whole image , i get incorrect results compared to looping that gives correct image.
test_prd.rehape(520,704). #incorrect results compared  row/column loop