Change the dimension of tensor - PyTorch Forums
https://discuss.pytorch.org/t/change-the-dimension-of-tensor/5145924.07.2019 · Yes, sure, First, the tensor a your provided has size [1, 4, 6] so unsqueeze(0) will add a dimension to tensor so we have now [1, 1, 4, 6]..unfold(dim, size, stride) will extract patches regarding the sizes. So first unfold will convert a to a tensor with size [1, 1, 2, 6, 2] and it means our unfold function extracted two 6x2 patches regarding the dimension with value 4.