After fixing `Shape`'s broadcastable information,
`local_subtensor_remove_broadcastable_index` started replacing all forms of
`shape(x)[0]` with `shape(x).dimshuffle(())` when `shape(x).broadcastable
== (True,)`, so a lot of the changes in this commit compensate for that
difference.