tf.io.deserialize_many_sparse
Десериализация и конкатенация SparseTensors из сериализованной мини-пачки.
tf.io.deserialize_many_sparse(
serialized_sparse, dtype, rank=None, name=None
)
Входной serialized_sparse должен быть матрицей строк типа [N x 3] формы [N x 3], где N — размер мини-пачки, а строки соответствуют упакованным результатам serialize_sparse. Ранги исходных объектов SparseTensor должны совпадать. При создании конечного SparseTensor его ранг на единицу выше, чем у входящих объектов SparseTensor (они были конкатенированы по новому измерению строки).
Значения формы выходного объекта SparseTensor для всех измерений, кроме первого, являются максимальными значениями среди входящих объектов SparseTensor для соответствующих измерений. Первое значение формы — это N, размер мини-пачки.
Индексы входных объектов SparseTensor предполагаются упорядоченными в стандартном лексикографическом порядке. Если это не так, после этого шага выполните sparse.reorder для восстановления порядка индексов.
Например, если входной сериализованный массив — это матрица [2, 3], представляющая два исходных объекта SparseTensor:
index = [ 0]
[10]
[20]
values = [1, 2, 3]
shape = [50]
и
index = [ 2]
[10]
values = [4, 5]
shape = [30]
то конечный десериализованный объект SparseTensor будет:
index = [0 0]
[0 10]
[0 20]
[1 2]
[1 10]
values = [1, 2, 3, 4, 5]
shape = [2 50]
| Аргументы | |
|---|---|
serialized_sparse | 2-мерный Tensor типа string с формой [N, 3]. Сериализованные и упакованные объекты SparseTensor. |
dtype | Ранг сериализованных объектов SparseTensor. |
rank | (необязательно) Целое число Python, ранг объектов SparseTensor. |
name | Префикс имени для возвращаемых тензоров (необязательно) |
| Возвращаемое значение | |
|---|---|
Объект SparseTensor, представляющий десериализованные объекты SparseTensor, конкатенированные по первому измерению объектов SparseTensor. Все сериализованные объекты |
© 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/io/deserialize_many_sparse