torch.unsqueeze#
- torch.unsqueeze(input, dim) Tensor #
返回一个新张量,并在指定位置插入一个大小为 1 的维度。
返回的张量与此张量共享相同的基础数据。
dim
的值可以在[-input.dim() - 1, input.dim() + 1)
范围内。负的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]])