When broadcast, the two trailing 1s line up, the middle 1 of the first
argument gets broadcast to match the 4 of the second argument, and
the “missing” 5 in the second argument gets broadcast to match the
leading 5 of the first argument. Thus your final shape is [5, 4, 1].