numpy squeezeの使い方
概要
- numpyのflattenと同じような働きをする
- axisを指定することで、指定した軸を削除することができる
サンプルコード
基本的な使い方
import numpy as np
x = np.array([[[1], [2], [3]]])
print(x.squeeze()) # [1 2 3]
print(x.squeeze(axis=0)) # [[1] [2] [3]]
要素が一つのとき、値を取り出す
import numpy as np
x = np.array([[12345]])
print(x.squeeze()[()]) # 12345