Rate this Page
★★★★★

torch.unsqueeze#

torch.unsqueeze(input,dim)→Tensor#

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

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

Adim value within the range[-input.dim()-1,input.dim()+1)can be used. Negativedim will correspond tounsqueeze()applied atdim =dim+input.dim()+1.

Parameters

  • input (Tensor) – the input tensor.

  • dim (int) – the index at which to insert the singleton dimension

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]])