Как установить высоту представления коллекции в соответствии с размером ячейки?

Question

У меня есть представление коллекции, в этом представлении коллекции я устанавливаю размер ячейки с помощью

func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, sizeForItemAt indexPath: IndexPath) -> CGSize {
    let collectionViewCellSize = collectionView.frame.width / 3 - 1
    return CGSize(width: collectionViewCellSize, height: collectionViewCellSize)

}

И я хочу установить высоту представления коллекции как высоту ячейки * 2 + 1

Как я могу сделать это?