numpy.squeeze()的用法

import numpy as np
x
= np.array([[[0], [1], [2]]]) print(x) """
x=
[[[0] [1] [2]]] """ print(x.shape) # (1, 3, 1) x1 = np.squeeze(x) # 从数组的形状中删除单维条目,即把shape中为1的维度去掉

print(x1) # [0 1 2] print(x1.shape) # (3,)

 

posted @ 2017-11-14 18:11  AI菌  阅读(51790)  评论(0编辑  收藏  举报