np.expand_dims()用于擴展數組的形狀

參數：

• values：數組
• axis：表示在該位置添加數據

用法示例：

注意數據擴展時 [] 所加的位置

import numpy as np a = np.array([[1,2,3],[4,5,6]]) print(a) print(a.shape)""" 輸出： [[1 2 3][4 5 6]] (2, 3) """ b = np.expand_dims(a, axis=0) print(b) print(b.shape)""" 輸出： [[[1 2 3][4 5 6]]] (1, 2, 3) """ b = np.expand_dims(a, axis=1) print(b) print(b.shape)""" 輸出： [[[1 2 3]][[4 5 6]]] (2, 1, 3) """ b = np.expand_dims(a, axis=2) print(b) print(b.shape)""" 輸出： [[[1][2][3]][[4][5][6]]] (2, 3, 1) """

總結

以上是生活随笔為你收集整理的np.expand_dims()介绍的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。