私は次のように計算の一連を有する仮説グラフ有する:不要になったときにテンソルをメモリから解放するには?
a_0 = tf.placeholder()
a_1 = some_op_1(a_0)
a_2 = some_op_2(a_1)
a_3 = some_op_3(a_2)
a_3
を計算するときことを確認します。 a_0
およびa_1
は不要であるため、a_3
のメモリを割り当てる前に破棄することができます。 TensorFlowにこのメモリ最適化を実行するよう依頼する方法はありますか(ある程度時間がかかることに同意します)。
これは、必要な場合にのみメモリを割り当てることに関するthisの質問と同じではありません。
編集:このネットワークは訓練されませんので、背景を心配しないでください。
これは自動的に行われます - 'a_3'計算が始まるときに、TensorFlowは' a_0'と 'a_1'をすでに破棄しています。 https://github.com/yaroslavvb/memory_utilを使用して、メモリ割り当て/割り当て解除のタイムラインを確認して、これが本当に正しいかどうかを確認することができます –