Made resultant shape preserve rank with ones in omitted axes