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()