在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)的二維數組。