在tensorflow输入处为预先训练的模型添加额外的图层

Question

我有facet模型（ckpt和元文件），它接受大小为(batch_size,160,160,3)的输入，并提供大小为(batch_size,128)的输出。

我的输入是一个k维向量，我有一个预处理函数（由一些卷积和池化层组成），它接受我的输入并给出(batch_size,160,160,3)输出。

如何在tensorflow中将此函数添加到训练模型并保存模型？

我是TF的新手。