現(xiàn)在我們有一個shape為(7352, 9, 128, 1)的numpy數(shù)組。
想要取出第2維的前三個數(shù)據(jù),構(gòu)成新數(shù)組(7352, 3, 128, 1)
我的思想是:將第2維數(shù)據(jù)轉(zhuǎn)置(transpose)到第一維,再用切片(slice)取出前三個數(shù)據(jù),再轉(zhuǎn)置回去:
print("# original", input.shape) input_transpose = input.transpose((1, 0, 2, 3)) print("# transpose", input_transpose.shape) input_slice = input_transpose[0:3] print("# slice", input_slice.shape) output = input_slice.transpose((1, 0, 2, 3)) print("# output", output.shape)
另外有需要云服務(wù)器可以了解下創(chuàng)新互聯(lián)scvps.cn,海內(nèi)外云服務(wù)器15元起步,三天無理由+7*72小時售后在線,公司持有idc許可證,提供“云服務(wù)器、裸金屬服務(wù)器、高防服務(wù)器、香港服務(wù)器、美國服務(wù)器、虛擬主機、免備案服務(wù)器”等云主機租用服務(wù)以及企業(yè)上云的綜合解決方案,具有“安全穩(wěn)定、簡單易用、服務(wù)可用性高、性價比高”等特點與優(yōu)勢,專為企業(yè)上云打造定制,能夠滿足用戶豐富、多元化的應(yīng)用場景需求。