前提・実現したいこと
Tensorflow 1.xx系で構築したグラフの内、指定したものを解放したい。
発生している問題
解放の方法が分からないため、教えていただきたいです。
該当のソースコード
以下は、質問用に作成したサンプルコードです。
・class tf_multiplyは、def create_graph_multiply(x)で値をx倍するグラフを構築して返します。
・サンプル中では2倍するグラフ、3倍するグラフを作成しています。
・2倍するグラフのみ開放したい場合、どのように記述すればよいですか?
python
1import tensorflow as tf 2 3class tf_multiply: 4 # 値を任意倍するグラフを作成するクラス 5 6 def __init__(self): 7 self.value = tf.placeholder(tf.float32) 8 9 def create_graph_multiply(self, x): 10 # 値をx倍するグラフを作成 11 result = self.value * x 12 return result 13 14if __name__ == "__main__": 15 16 # ---グラフ生成フェーズ 17 cls_obj = tf_multiply() 18 19 # 値を2倍するグラフ 20 multiply2 = cls_obj.create_graph_multiply(2) 21 22 # 値を3倍するグラフ 23 multiply3 = cls_obj.create_graph_multiply(3) 24 25 26 # ---計算フェーズ 27 sess = tf.Session() 28 print(sess.run(multiply2, feed_dict={cls_obj.value:3})) 29 # 6.0 30 print(sess.run(multiply3, feed_dict={cls_obj.value:3})) 31 # 9.0 32 33 # ---解放フェーズ 34 35 # 以下のように指定したグラフを解放したい 36 # del multiply2 ????? 37 38
補足情報
Tensorflow: ver1.15
あなたの回答
tips
プレビュー