linalg:expand-dims
(linalg:expand-dims array axis)
array の axis の位置に extent 1 の軸を挿入したコピーを返します (numpy の np.expand_dims、torch の unsqueeze)。負の axis は結果の末尾から数えるので、-1 は新しい軸を末尾に追加します。行優先の要素の並びは変わらず形状だけが変わり、要素幅は入力に従います。逆操作は linalg:squeeze です。
(linalg:expand-dims array axis)
array の axis の位置に extent 1 の軸を挿入したコピーを返します (numpy の np.expand_dims、torch の unsqueeze)。負の axis は結果の末尾から数えるので、-1 は新しい軸を末尾に追加します。行優先の要素の並びは変わらず形状だけが変わり、要素幅は入力に従います。逆操作は linalg:squeeze です。