Esempi in Python per lookup_table_insert_v2

Esempio n. 1

File: boosted_trees.py Progetto: zhongleiwang/tensorflow-1

 def insert(self, tree_ids, node_ids, logits):
     """Inserts values and returns the op."""
     insert_op = lookup_ops.lookup_table_insert_v2(
         self._table_ref, self._example_ids,
         array_ops.concat([
             array_ops.expand_dims(
                 array_ops.bitcast(tree_ids, dtypes.float32), 1),
             array_ops.expand_dims(
                 array_ops.bitcast(node_ids, dtypes.float32), 1),
             logits,
         ],
                          axis=1,
                          name='value_concat_for_cache_insert'))
     return insert_op

Esempio n. 2

File: boosted_trees.py Progetto: Huoxubeiyin/tensorflow

 def insert(self, tree_ids, node_ids, logits):
   """Inserts values and returns the op."""
   insert_op = lookup_ops.lookup_table_insert_v2(
       self._table_ref, self._example_ids,
       array_ops.concat(
           [
               array_ops.expand_dims(
                   array_ops.bitcast(tree_ids, dtypes.float32), 1),
               array_ops.expand_dims(
                   array_ops.bitcast(node_ids, dtypes.float32), 1),
               logits,
           ],
           axis=1,
           name='value_concat_for_cache_insert'))
   return insert_op