阅读量:1
在numpy中，unsqueeze函数用于在指定维度上增加一个维度。具体用法如下：
numpy.unsqueeze(arr, axis) 参数说明：
- arr：输入的数组或矩阵。
- axis：要在其位置插入新维度的维度索引。
实际上，unsqueeze函数是使用reshape函数实现的。它可以在指定的维度上插入一个大小为1的维度，从而增加数组或矩阵的维度。
示例：
import numpy as np # 创建一个一维数组 a = np.array([1, 2, 3, 4, 5]) # 使用unsqueeze在维度0上增加一个维度 b = np.unsqueeze(a, axis=0) print(b.shape) # 输出(1, 5) # 使用unsqueeze在维度1上增加一个维度 c = np.unsqueeze(a, axis=1) print(c.shape) # 输出(5, 1) 在上述示例中，通过unsqueeze函数在维度0上增加了一个维度，将原本形状为(5,)的一维数组变为形状为(1, 5)的二维数组。然后，在维度1上增加了一个维度，将原本的一维数组变为(5, 1)的二维数组。
