def lookup(self, keys, name=None): """Looks up `keys` in a table, outputs the corresponding values. The `default_value` is used for keys not present in the table. Args: keys: Keys to look up. Can be a tensor of any shape. Must match the table's key_dtype. name: A name for the operation (optional). Returns: A tensor containing the values in the same shape as `keys` using the table's value type. Raises: TypeError: when `keys` do not match the table data types. """ if keys.dtype != self._key_dtype: raise TypeError( "Signature mismatch. Keys must be dtype %s, got %s." % (self._key_dtype, keys.dtype)) with ops.name_scope(name, "%s_lookup_table_find" % self._name, [self._table_ref, keys]) as name: # pylint: disable=protected-access values = gen_data_flow_ops._lookup_table_find(self._table_ref, keys, self._default_value, name=name) if keys.get_shape().ndims is not None and keys.get_shape().ndims > 0: values.set_shape( tensor_shape.TensorShape([keys.get_shape().dims[0] ]).concatenate(self._value_shape)) return values
def lookup(self, keys, name=None): """Returns the values for the given 'keys' tensor. If an element on the key tensor is not found in the table, the default_value is used. Args: keys: The tensor for the keys. name: Optional name for the op. Returns: The operation that looks up the keys. Raises: TypeError: when 'keys' or 'default_value' doesn't match the table data types. """ if name is None: name = "%s_lookup_table_find" % self._name if keys.dtype != self._key_dtype: raise TypeError( "Signature mismatch. Keys must be dtype %s, got %s." % (self._key_dtype, keys.dtype)) return gen_data_flow_ops._lookup_table_find(self._table_ref, keys, self._default_value, name=name)
def lookup(self, keys, name=None): """Looks up `keys` in a table, outputs the corresponding values. The `default_value` is use for keys not present in the table. Args: keys: Keys to look up. name: Optional name for the op. Returns: The operation that looks up the keys. Raises: TypeError: when `keys` or `default_value` doesn't match the table data types. """ if name is None: name = "%s_lookup_table_find" % self._name if keys.dtype != self._key_dtype: raise TypeError("Signature mismatch. Keys must be dtype %s, got %s." % ( self._key_dtype, keys.dtype)) return gen_data_flow_ops._lookup_table_find( self._table_ref, keys, self._default_value, name=name)
def lookup(self, keys, name=None): """Looks up `keys` in a table, outputs the corresponding values. The `default_value` is used for keys not present in the table. Args: keys: Keys to look up. Can be a tensor of any shape. Must match the table's key_dtype. name: A name for the operation (optional). Returns: A tensor containing the values in the same shape as `keys` using the table's value type. Raises: TypeError: when `keys` do not match the table data types. """ if keys.dtype != self._key_dtype: raise TypeError("Signature mismatch. Keys must be dtype %s, got %s." % (self._key_dtype, keys.dtype)) with ops.name_scope(name, "%s_lookup_table_find" % self._name, [self._table_ref, keys]) as name: # pylint: disable=protected-access values = gen_data_flow_ops._lookup_table_find( self._table_ref, keys, self._default_value, name=name) if keys.get_shape().ndims is not None and keys.get_shape().ndims > 0: values.set_shape( tensor_shape.TensorShape([keys.get_shape().dims[0]]).concatenate( self._value_shape)) return values
def lookup(self, keys, name=None): """Returns the values for the given 'keys' tensor. If an element on the key tensor is not found in the table, the default_value is used. Args: keys: The tensor for the keys. name: Optional name for the op. Returns: The operation that looks up the keys. Raises: TypeError: when 'keys' or 'default_value' doesn't match the table data types. """ if name is None: name = "%s_lookup_table_find" % self._name if keys.dtype != self._key_dtype: raise TypeError("Signature mismatch. Keys must be dtype %s, got %s." % ( self._key_dtype, keys.dtype)) return gen_data_flow_ops._lookup_table_find( self._table_ref, keys, self._default_value, name=name)
def lookup(self, keys, name=None): """Looks up `keys` in a table, outputs the corresponding values. The `default_value` is used for keys not present in the table. Args: keys: Keys to look up. May be either a `SparseTensor` or dense `Tensor`. name: A name for the operation (optional). Returns: A `SparseTensor` if keys are sparse, otherwise a dense `Tensor`. Raises: TypeError: when `keys` or `default_value` doesn't match the table data types. """ if name is None: name = "%s_lookup_table_find" % self._name key_tensor = keys if isinstance(keys, ops.SparseTensor): key_tensor = keys.values if keys.dtype != self._key_dtype: raise TypeError("Signature mismatch. Keys must be dtype %s, got %s." % (self._key_dtype, keys.dtype)) # pylint: disable=protected-access values = gen_data_flow_ops._lookup_table_find(self._table_ref, key_tensor, self._default_value, name=name) # pylint: enable=protected-access values.set_shape(key_tensor.get_shape()) if isinstance(keys, ops.SparseTensor): return ops.SparseTensor(keys.indices, values, keys.shape) else: return values
def lookup(self, keys, name=None): """Looks up `keys` in a table, outputs the corresponding values. The `default_value` is used for keys not present in the table. Args: keys: Keys to look up. May be either a `SparseTensor` or dense `Output`. name: A name for the operation (optional). Returns: A `SparseTensor` if keys are sparse, otherwise a dense `Output`. Raises: TypeError: when `keys` or `default_value` doesn't match the table data types. """ if name is None: name = "%s_lookup_table_find" % self._name key_tensor = keys if isinstance(keys, sparse_tensor.SparseTensor): key_tensor = keys.values if keys.dtype != self._key_dtype: raise TypeError( "Signature mismatch. Keys must be dtype %s, got %s." % (self._key_dtype, keys.dtype)) # pylint: disable=protected-access values = gen_data_flow_ops._lookup_table_find(self._table_ref, key_tensor, self._default_value, name=name) # pylint: enable=protected-access values.set_shape(key_tensor.get_shape()) if isinstance(keys, sparse_tensor.SparseTensor): return sparse_tensor.SparseTensor(keys.indices, values, keys.shape) else: return values
def lookup(self, keys, default): return gen_data_flow_ops._lookup_table_find(self.table_ref, keys, default)