UICollectionView cell with constant heigh and dynamic width depending on UIImageView width

Question

How can I set the collection view cell width depending on image view. I have tried with the give code below but the width is same for all cell. I have set the height constra