在numpy中,unsqueeze函数用于在指定维度上增加一个维度。具体用法如下:
numpy.unsqueeze(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)的二维数组。