Блог

Я пытаюсь немного поработать с ручным управлением памятью и не уверен, как переместить тензор на процессор после завершения операции. Для моего варианта использования я хотел бы запустить STFT на GPU, а затем немедленно переместить тензор на CPU, потому что он занимает огромный объем памяти. Что-то вроде.

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

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