Numpyのunsqueeze関数の目的は何ですか?

numpyのunsqueeze関数は指定した次元に新しい次元を追加する機能です。具体的には、この関数は入力配列の次元を1つ拡張し、より高次元の配列に変換します。

unsqueeze関数の構文は次のようになります:

numpy.unsqueeze(a, axis)

aというのは入力配列を表し、axisは新しい次元をどの軸に追加するかを示す。axisパラメーターの取りうる値は[-(n+1), n+1]の範囲にあり、ここでnはaの次元数を表す。axisパラメーターが負の数の場合、それは配列の最後の次元を基準とすることを意味する。

unsqueeze関数は元の配列の中身を変更しませんが、新しい配列を返します。新しい配列の次元は1つ増え、その長さは1になります。

例えば、2次元配列aがあるとします。

a = np.array([[1, 2], [3, 4]])

unsqueeze関数を使用して、2番目の次元に新しい次元を追加することができます。

b = np.unsqueeze(a, 1)

現時点で、配列bの次元は(2、1、2)であり、元の配列には2番目の次元に長さ1の次元が追加されています。

bannerAds