In the case of a default axis argument and a 1-d tensor, don't create
flatten and reshape nodes that need to be optimized out when it's simple
enough not to do so.