溫馨提示×

numpy中unsqueeze的用法是什么

小億
383
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
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女