本当にちょっとしたミスでmat_mul関数内でiの所にjを書いてしまって
いました。本当にすいませんでした。