Pytorch 擴充tensor次元

1、擴充1維

input = torch.randn(256, 192, 192)
input = torch.unsqueeze(input, 0)
           

2、擴充多元

input = torch