fix the bug : RuntimeError: mat1 and mat2 shapes cannot be multiplied 4e6143b zhengr commited on Jul 27, 2023