使用Swift 3在展开视图中设置动态表格单元格高度

Question

我正在我的应用中进行折叠/展开表格视图。我在视图控制器中添加表视图并编写用于折叠/展开的代码。

func tableView(_ tableView: UITableView, heightForRowAtIndexPath indexPath: NSIndexPath) -> CGFloat {
    return sections[indexPath.section].collapsed! ? 0 : 44
}

当表格行折叠时，它会变为“0”并且展开高度将为44。
我不想设置44，因为在每个表格行中，我将“查看”卡片设计与图像放在一起。
我希望根据查看卡的数量来动态高度。
我写了 UITableViewAutomaticDimension 而不是44，但它不起作用。
请帮帮我如何实现？