いくつかの操作のnumpyの変更ndarray寸法

np.expand_dims(a, axis)- 挿入軸寸法指定された入力配列の配列の拡張形状、新しい寸法がAであります

例:

>>>x = np.array([1,2])
>>>x.shape
(2,)
>>>y = np.expand_dims(x, axis=0)
>>>y
array([[1,2]])
>>>y.shape
(1,2)
>>>y = np.expand_dims(x, axis=1)
>>>y
array([[1],
       [2]])
>>>y.shape
(2,1)

numpy.squeeze(a, axis=None)- 寸法で削除配列は次元1で、軸が寸法を除去するために指定することができ、アレイは、これらの寸法を除去した後の寸法1戻されます

例:

>>>x = np.array([[[0], [1], [2]]]
>>>x.shape
(1, 3, 1)
>>>np.squeeze(x).shape
(3,)
>>>np.squeeze(x, axis=0).shape
(3,1)

したがって、スクイズとexpand_dimsの1組逆数演算

おすすめ

転載: www.cnblogs.com/patrolli/p/11992198.html