np.expand_dims()介绍

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()介绍的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。