torch.unsqueeze()和torch.squeeze()函数如何使用

torch.unsqueeze()函数介绍

torch.unsqueeze()函数可以在给定位置插入一个维度，主要用于增加维度，使其可以满足某些操作的要求。

torch.unsqueeze()函数使用

1. torch.unsqueeze()函数的语法为：torch.unsqueeze(input,dim)，其中input表示输入的张量，dim表示插入维度的位置，可以是0、1、2等数字，也可以是负数，如果输入是负数，则表示从末尾开始插入维度。

2. 下面是一个例子：

import torchx = torch.randn(2,3)y = torch.unsqueeze(x, 0)print(y.size())# torch.Size([1, 2, 3])

在上面的代码中，x是一个2*3的张量，dim参数设定为0，表示在索引0的位置插入一个新的维度，所以y的size为torch.Size([1, 2, 3])。

torch.squeeze()函数介绍

torch.squeeze()函数可以删除张量中所有维度值为1的维度，主要用于减少维度，使其可以满足某些操作的要求。

torch.squeeze()函数使用

1. torch.squeeze()函数的语法为：torch.squeeze(input)，其中input表示输入的张量，可以是任意维度的张量。

2. 下面是一个例子：

import torchx = torch.randn(1,2,1,3)y = torch.squeeze(x)print(y.size())# torch.Size([2, 3])

在上面的代码中，x是一个1*2*1*3的张量，使用torch.squeeze()函数可以删除维度值为1的维度，所以y的size为torch.Size([2, 3])。