Ability to sort dataset batch to reduce exccessive padding?

Thanks! I found some good ideas here - Tensorflow-esque bucket by sequence length