torch:mean

(torch:mean a &key axis keepdims)

微分可能な算術平均 (linalg:mean、:axis / :keepdims の規則は torch:sum と同じ) です。backward は合計の随伴を縮約した要素数で割ったものです。torch:sub の二乗の平均が MSE 損失になります。