tf.compat.v1.sparse_merge
Объединяет пакет идентификаторов признаков и значений в один SparseTensor. (устарела)
tf.compat.v1.sparse_merge(
sp_ids, sp_values, vocab_size, name=None, already_sorted=False
)
Чаще всего эта функция используется, когда идентификаторы признаков и соответствующие им значения хранятся на диске в протоколах Example. parse_example возвращает пакет идентификаторов и пакет значений, а эта функция объединяет их в одну логическую SparseTensor для использования в таких функциях, как sparse_tensor_dense_matmul, sparse_to_dense и т. д.
Функция SparseTensor возвращает объект со следующими свойствами:
-
indicesэквивалентноsp_ids.indices, у которого последняя размерность отброшена и заменена наsp_ids.values. -
values— это простоsp_values.values. - Если
sp_ids.dense_shape = [D0, D1, ..., Dn, K], тоoutput.shape = [D0, D1, ..., Dn, vocab_size].
Например, рассмотрим следующие векторы признаков:
vector1 = [-3, 0, 0, 0, 0, 0] vector2 = [ 0, 1, 0, 4, 1, 0] vector3 = [ 5, 0, 0, 9, 0, 0]
Они могут храниться в разреженном виде в следующих протобуфах Example: сохраняются только идентификаторы признаков (номера столбцов, если рассматривать векторы как матрицу) ненулевых элементов и соответствующие им значения:
examples = [Example(features={
"ids": Feature(int64_list=Int64List(value=[0])),
"values": Feature(float_list=FloatList(value=[-3]))}),
Example(features={
"ids": Feature(int64_list=Int64List(value=[1, 4, 3])),
"values": Feature(float_list=FloatList(value=[1, 1, 4]))}),
Example(features={
"ids": Feature(int64_list=Int64List(value=[0, 3])),
"values": Feature(float_list=FloatList(value=[5, 9]))})]
Вызов parse_example для этих примеров создаст словарь с записями для "ids" и "values". Если передать эти два объекта в эту функцию вместе с vocab_size=6, она вернет SparseTensor, в разреженном виде представляющий все три экземпляра. А именно, свойство indices будет содержать координаты ненулевых элементов матрицы признаков (первая размерность — номер строки матрицы, то есть индекс в пакете, а вторая — номер столбца, то есть идентификатор признака); values будет содержать фактические значения. shape будет формой исходной матрицы, то есть (3, 6). Для приведенного выше примера результат будет равен:
SparseTensor(indices=[[0, 0], [1, 1], [1, 3], [1, 4], [2, 0], [2, 3]],
values=[-3, 1, 4, 1, 5, 9],
dense_shape=[3, 6])
Этот метод обобщается на размерности выше двух: достаточно передать списки и для sp_ids, и для vocab_size. В этом случае результирующий SparseTensor будет иметь следующие свойства:
-
indicesэквивалентноsp_ids[0].indices, у которого последняя размерность отброшена и объединена сsp_ids[0].values, sp_ids[1].values, .... -
values— это простоsp_values.values. - Если
sp_ids.dense_shape = [D0, D1, ..., Dn, K], тоoutput.shape = [D0, D1, ..., Dn] + vocab_size.
| Аргументы | |
|---|---|
sp_ids | Один SparseTensor со свойством values типа int32 или int64, список Python из таких SparseTensor или список таких списков. |
sp_values | SparseTensor любого типа. |
vocab_size | Скалярный тензор int64 (или целое число Python), содержащий новый размер последней размерности, all(0 <= sp_ids.values < vocab_size). Либо список таких значений, где all(0 <= sp_ids[i].values < vocab_size[i]) задано для всех i. |
name | Префикс имени для возвращаемых тензоров (необязательно) |
already_sorted | Логическое значение, указывающее, отсортированы ли уже значения для каждого элемента пакета в sp_values. Если да, сортировка пропускается. По умолчанию — False (необязательно). |
| Возвращает | |
|---|---|
SparseTensor, компактно представляющий пакет идентификаторов признаков и значений; его удобно передавать функциям, ожидающим такой SparseTensor. |
| Вызывает исключения | |
|---|---|
TypeError | Если sp_values не является SparseTensor. Или если sp_ids — это не SparseTensor и не список таких объектов. Или если vocab_size не является Tensor или целым числом Python, а sp_ids является SparseTensor. Или если vocab_size не является списком или списком таких объектов, а sp_ids является списком. |
ValueError | Если sp_ids и vocab_size — списки разной длины. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/compat/v1/sparse_merge