Вы можете попробовать что-то вроде этого.
func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, sizeForItemAt indexPath: IndexPath) -> CGSize {
return CGSize(width: collectionView.frame.width, height: collectionView.frame.height)
}
Редактировать: Вы должны подтвердить UICollectionViewDelegateFlowLayout
Делегат, чтобы использовать эту функцию.