tf.io.deserialize_many_sparse
| Просмотреть исходный код на GitHub |
Десериализация и конкатенация SparseTensors из сериализованной минипачки.
tf.io.deserialize_many_sparse(
serialized_sparse, dtype, rank=None, name=None
)
Вход serialized_sparse должен быть матрицей строк типа [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/versions/r2.9/api_docs/python/tf/io/deserialize_many_sparse