Python tf_get_sessionの例

コード例 #1

ファイル: __init__.py プロジェクト: MLDL/tensorfuse

 def __call__(self, *args):
     session = compat.tf_get_session()
     compat.tf_ensure_init_variables()
     try:
         if self._update_op:
             output_vals = session.run(
                 self._output_list + [self._update_op], feed_dict=dict(zip(self._inputs, args)))[:-1]
         else:
             output_vals = session.run(
                 self._output_list, feed_dict=dict(zip(self._inputs, args)))
         if isinstance(self._outputs, (list, tuple)):
             return output_vals
         else:
             return output_vals[0]
     except Exception as e:
         import traceback
         traceback.print_exc()
         import ipdb
         ipdb.set_trace()

コード例 #2

 def __call__(self, *args):
     session = compat.tf_get_session()
     compat.tf_ensure_init_variables()
     try:
         if self._update_op:
             output_vals = session.run(
                 self._output_list + [self._update_op],
                 feed_dict=dict(zip(self._inputs, args)))[:-1]
         else:
             output_vals = session.run(self._output_list,
                                       feed_dict=dict(
                                           zip(self._inputs, args)))
         if isinstance(self._outputs, (list, tuple)):
             return output_vals
         else:
             return output_vals[0]
     except Exception as e:
         import traceback
         traceback.print_exc()
         import ipdb
         ipdb.set_trace()

コード例 #3

ファイル: api.py プロジェクト: spankmasters/tensorfuse

def _tf_variable_set_value(self, val, borrow=None):
    tf_get_session().run(tf.assign(self, val))

コード例 #4

ファイル: api.py プロジェクト: spankmasters/tensorfuse

def _tf_variable_get_value(self, borrow=None):
    tf_ensure_init_variables()
    return tf_get_session().run(self)

コード例 #5

ファイル: api.py プロジェクト: MLDL/tensorfuse

def _tf_variable_set_value(self, val, borrow=None):
    tf_get_session().run(tf.assign(self, val))

コード例 #6

ファイル: api.py プロジェクト: MLDL/tensorfuse

def _tf_variable_get_value(self, borrow=None):
    tf_ensure_init_variables()
    return tf_get_session().run(self)