numpy中的unsqueeze函數的作用是在指定的維度上增加一個新的維度。具體來說,它會將輸入數組的維度擴展一維,使其變成一個更高維度的數組。
unsqueeze函數的語法如下:
numpy.unsqueeze(a, axis)
其中,a表示輸入的數組,axis表示要在哪個維度上增加新的維度。axis參數的取值范圍為[-(n+1), n+1],其中n為a數組的維度數。如果axis參數為負數,則表示相對于數組的末尾維度的偏移。
需要注意的是,unsqueeze函數不會改變輸入數組的內容,而是返回一個新的數組。新數組的維度會增加一維,其長度為1。
舉個例子,假設有一個二維數組a:
a = np.array([[1, 2], [3, 4]])
可以使用unsqueeze函數在第二個維度上增加一個新的維度:
b = np.unsqueeze(a, 1)
此時,數組b的維度為(2, 1, 2),即原數組在第二個維度上增加了一個長度為1的維度。