np.expand_dims()介绍
Posted K同学啊
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了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()介绍的主要内容，如果未能解决你的问题，请参考以下文章
python灵活使用np.expand_dims np.tiley