改了好久都没有成功，跪求大佬指点

估计是numpy的形状问题，最后面写了