np.expand_dims:用于扩展数组的形状
原始数组：import numpy as npIn :a = np.array([[,]])a.shapeOut:(1, 2, 3)
np.expand_dims(a, axis=0)表示在0位置添加数据,转换结果如下：
In :b = np.expand_dims(a, axis=0)bOut:array([[[, ]]])In :b.shapeOut:(1, 1, 2, 3)
np.expand_dims(a, axis=1)表示在1位置添加数据,转换结果如下：
In :c = np.expand_dims(a, axis=1)cOut:array([[[, ]]])In :c.shapeOut:(1, 1, 2, 3)
np.expand_dims(a, axis=2)表示在2位置添加数据,转换结果如下：
In :d = np.expand_dims(a, axis=2)dOut:array([[[], []]])In :d.shapeOut:(1, 2, 1, 3)
np.expand_dims(a, axis=3)表示在3位置添加数据,转换结果如下：
In :e = np.expand_dims(a, axis=3)eIn :e.shapeOut:(1, 2, 3, 1)
能在(1,2,3)中插入的位置总共为4个，再添加就会出现以下的警告，要不然也会在后面某一处提示AxisError。
In :f = np.expand_dims(a, axis=4)D:\ProgramData\Anaconda3\envs\dlnd\lib\site-packages\ipykernel_launcher.py:1: DeprecationWarning: Both axis > a.ndim and axis < -a.ndim - 1 are deprecated and will raise an AxisError in the future."""Entry point for launching an IPython kernel.
作者：caoqi95
链接：https://www.jianshu.com/p/da10840660cb
来源：简书
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。
页:
[1]