我无法理解PyTorch 文档中的示例如何对应于解释:
返回一个新的张量,其尺寸为 1,插入到指定位置。[...] >>> 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]]) Run Code Online (Sandbox Code Playgroud)
返回一个新的张量,其尺寸为 1,插入到指定位置。[...]
>>> 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]])
python torch pytorch
python ×1
pytorch ×1
torch ×1