tf_util.py
520 Bytes
import tensorflow as tf
class tf_base(object):
def __init__(self,graph=None):
if graph is None:
self.graph = tf.Graph()
else:
self.graph = graph
def initialize(self):
self.session = tf.Session(graph=self.graph)
with self.graph.as_default():
self.init_variable = tf.global_variables_initializer()
self.run(self.init_variable)
def run(self,op,feed_dict=None,options=None,run_metadata=None):
return self.session.run(op,feed_dict,options,run_metadata)
def close(self):
self.session.close()