Dynami c высота UICollectionView с максимальной шириной

Question

Я разрабатываю Tagview с использованием CollectionView, и один тег имеет несколько строк. Но я не могу этого достичь.

Я пытался по приведенному ниже методу

override func preferredLayoutAttributesFitting(_ layoutAttributes: UICollectionViewLayoutAttributes) -> UICollectionViewLayoutAttributes {
    setNeedsLayout()
    layoutIfNeeded()
    let size = contentView.systemLayoutSizeFitting(layoutAttributes.size)
    var newFrame = layoutAttributes.frame
    newFrame.size.height = ceil(size.height)
    layoutAttributes.frame = newFrame
    return layoutAttributes  
}

С помощью вышеуказанного метода я могу увеличить высоту, но ширина остается c исходя из расчетного размера предоставь. Можем ли мы найти решение, в котором ширина должна быть меньше, чем ширина коллекции, а высота должна быть динамической c?

Пожалуйста, обратитесь к изображению, чтобы решить проблему, с которой я столкнулся.

Спасибо за аванс Рам