numpy矩陣軸轉換(transpose,)

輸入數據shape是(70,66,3)
我需要的shape是(3,70,66)
這時候可以用np提供的transpose函數,
data = data.transpose((2, 0, 1))

表明將第2位的移動到第0位,以次類推

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章