軸の追加と削除
次のコードを実行したときの出力として、正しいものを1つ選べ。
import numpy as np
a = np.expand_dims(np.array([1, 2]), 0)
print(a.shape, np.squeeze(a).shape)- A
(2, 1) (2,) - B
(1, 2) (1, 2) - C
(1, 2) (2,) - D
(2,) (1, 2)
正解と解説を見る
正解：C
np.expand_dims(配列, 軸) で軸を追加し、np.squeeze で大きさ 1 の軸を取り除く。ミニバッチの次元の調整などに使う。
| 順 | 実行する行 | 実行後の変数 |
|---|---|---|
| 1 | `import numpy as np` | — |
| 2 | `a = np.expand_dims(np.array([1, 2]), 0)` | a = [[1, 2]] |
| 3 | `print(a.shape, np.squeeze(a).shape)` | a = [[1, 2]] |
| 結果 | 実際の出力 | (1, 2) (2,) |
- A
誤り。軸 0 に追加している。
- B
誤り。
squeezeで軸が取り除かれる。 - C
正しい。
expand_dimsで大きさ 1 の軸を追加し、squeezeで大きさ 1 の軸を取り除く。 - D
誤り。
expand_dimsで軸が増える。
このサイトの問題は、各試験の公式に示された出題範囲に沿って作ったオリジナルの模擬問題です。本試験の問題ではありません。一般社団法人Pythonエンジニア育成推進協会の公式サービスではありません。