torch.unsqueeze¶
- torch.unsqueeze(input, dim) Tensor ¶
返回一个新张量,在指定位置插入一个大小为一的维度。
返回的张量与该张量共享相同的底层数据。
可以使用范围为
[-input.dim() - 1, input.dim() + 1)
内的dim
值。负dim
将对应于应用于dim
=dim + input.dim() + 1
的unsqueeze()
。示例
>>> 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]])