softmax expand (4 dim)