Allow multiple variable-rank dimension specs in check shapes

Allow multiple variable-rank dimension specs in check shapes: a task in MiMo-V2.6-RL-harbor-code: MiMo-V2.6-RL Code (Harbor) (Harbor dataset). This is hard for reasons similar to why GPflow/check shapes 5 is hard. In general, if there are multiple variable-rank dimensions in one specification, we…