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