torch.unsqueeze¶
- torch.unsqueeze(input, dim) Tensor ¶
返回一个新的张量,其中在指定位置插入大小为 1 的维度。
返回的张量与此张量共享相同的底层数据。
可以使用范围
[-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]])