We do not allow dynamic shape in model, so currently this is impossible to do.
One potential work around is having a fix shape alpha with a shape is equal to the max shape (5 in this case), and dynamically assign zeros to some element. But again, inferencing these kind of model is also very tricky.