如何分析张量流内存?

5 python tensorflow

不,感谢精彩的张量流分析器文档显示功能,而不是实际的使用方式。我遇到了内存泄漏,正如我之前在帖子中描述的那样。tf.function从那时起,我一直在努力寻找一种方法来记录内部发生的事情的记忆。但正如一直以来的情况一样,文档就像从来不存在一样毫无用处。对于那些在使用时经历过内存泄漏的人来说tf.function,有没有办法实际查看哪些内存分配给了哪些对象?忘记那些不起作用的tracemalloc、objgraph、内存分析器,当然还有没有文档的官方张量流分析器。假设除非有某个地方,否则泄漏不会发生tf.function,这里有哪些选择?