温馨提示×

numpy中unsqueeze的用法是什么

小亿
221
2024-01-10 10:34:35
栏目: 编程语言

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

0