numpy中unsqueeze的用法是什么

在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)的二维数组。