torch:unsqueeze
(torch:unsqueeze a axis)
微分可能な広がり 1 の軸の挿入 (linalg:expand-dims、torch の unsqueeze) です。負の軸は結果の末尾から数えるため -1 は末尾に追加します。行優先の並びは変わらないので backward は reshape です。
(torch:unsqueeze a axis)
微分可能な広がり 1 の軸の挿入 (linalg:expand-dims、torch の unsqueeze) です。負の軸は結果の末尾から数えるため -1 は末尾に追加します。行優先の並びは変わらないので backward は reshape です。