"batch samples features, out features -> batch samples out" · the shared axis is multiplied away, the private axes survive