怎么解决呢，带有下标的赋值维度不匹配。

你一步一步的写，然后每一步用 xx.shape 看一下维度就行。有时候维度的问题确实比较头疼

用assert进行约束维度