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