-
torch.squeeze()
torch.squeeze(input, dim=None, out=None)参数:
- input (Tensor) – 输入张量
- dim (int, optional) – 如果给定，则
input只会在给定维度挤压 - out (Tensor, optional) – 输出张量
注意： 返回张量与输入张量共享内存，所以改变其中一个的内容会改变另一个。
-
torch.unsqueeze()
返回一个新的张量，对输入的制定位置插入维度 1
注意： 返回张量与输入张量共享内存，所以改变其中一个的内容会改变另一个。
如果dim为负，则将会被转化dim+input.dim()+1dim+input.dim()+1
参数:
- tensor (Tensor) – 输入张量
- dim (int) – 插入维度的索引
- out (Tensor, optional) – 结果张量
