If the condition and argument types are compatible, i.e. there is no broadcast, the resulting shape should be the shape of the arguments, not the shape of the condition.