torch:squeeze

(torch:squeeze a &key axis)

微分可能な広がり 1 の軸の除去 (linalg:squeeze) です。:axis なしではすべて、指定すると名指しした軸 (またはそのリスト) だけを除去します。すべての軸を除去するとスカラーテンソルになります。