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

參數(shù)：

• values：數(shù)組
• axis：表示在該位置添加數(shù)據(jù)

用法示例：

注意數(shù)據(jù)擴展時 [] 所加的位置

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) """

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。