Python Tensorflow：将tensor从GPU移动到CPU以释放内存

我正在尝试进行一些手动内存管理，不确定一旦操作完成，如何将张量移动到CPU上。对于我的用例，我希望在GPU上运行STFT，然后立即将张量移动到CPU上，因为它占用大量内存。差不多

with tf.device("GPU:0"):
    foo = tf.signal.stft(bar, ...)

with tf.device("CPU:0"):
    baz = foo
del foo