正則化タームの単位行列の次元を保持している変数で指定するように変更した。