dimensions that are broadcastable.

Also, a small modif to make those assert always work. Add a test for this.

This caused a crahs in the optimizer, as in one case, we didn't know
the shape was broadcastable by that path.