einsumなんてそもそも計算領空演算でもないし、それ単体で最適化すること自体微妙だわ