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])
Python

在上面的代码中，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])
Python

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