Solved by AbdulsalamBande in post #2 
              
              
                Hy man, you can try the below 
import torch.nn.functional as F
data = torch.ones((64, 3, 240, 320))
# pad(left, right, top, bottom)
new_data = F.pad(input=data, pad=(0, 1, 0, 0), mode='constant', value=0)
print(new_data)

Reference reshaping a tensor with padding in pytorch