Weight Must Be 2d Shapes