请问tensorflow如何查看一个神经网络运行时所占用的内存大小?

tensorflow有没有专门的函数可以查看神经网络运行时所占用的内存,如果没有的话有什么方法可以查看呢?

跪求大佬们的指导

不知道你这个问题是否已经解决, 如果还没有解决的话:

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^