在Tensorflow中创建张量及其所有依赖项的副本

Question

我想编写一个函数，它接收一个logits层作为输入，并复制它及其所有依赖的张量和操作。

换句话说，给定一些图的logits层：

我想把它变成这个（不知道网络的架构）：

有什么建议吗？

感谢。