首页 > 解决方案 > 从 2D tensorflow 变量中删除单行(GPU 上的可用内存)

问题描述

我有一个 NxM 大小的张量流变量存储在 GPU 中。在某些时候,我想删除一些行,即释放它们在 GPU 中占用的内存。我不想通过执行sess.close()或删除会话来释放所有内存资源。这在张量流中是否可以从变量中选择随机行并完全擦除它?我已经阅读有关释放内存的答案,但它是针对张量流操作的。

标签: pythontensorflow

解决方案


推荐阅读