torch.unsqueeze

torch.unsqueeze(input, dim) -> Tensor

Returns a new tensor with a dimension of size one inserted at the specified position.

The returned tensor shares the same underlying data with this tensor.

A dim value within the range [-input.dim() - 1, input.dim() + 1) can be used. Negative dim will correspond to unsqueeze() applied at dim = dim + input.dim() + 1.

주어진 input에 dim으로 지정된 위치에 dimension size를 하나 추가한 tensor를 반환한다.
반환된 tensor는 input과 같은 data를 가지고 있다. [input.dim-1, input.dim+1) 사이의 값이 dim으로 쓰일 수 있다. dim이 음수인 경우 dim = dim+input.dim()+1 의 값으로 unsqueeze하는 것에 대응된다.

• example

>>> x = torch.tensor([1, 2, 3, 4])
>>> torch.unsqueeze(x, 0)
tensor([[ 1,  2,  3,  4]])
>>> torch.unsqueeze(x, 1)
tensor([[ 1],
        [ 2],
        [ 3],
        [ 4]])

Reference

1. https://pytorch.org/docs/stable/generated/torch.unsqueeze.html