tf.raw_ops.TakeManySparseFromTensorsMap
Читает SparseTensors из SparseTensorsMap и объединяет их.
tf.raw_ops.TakeManySparseFromTensorsMap(
sparse_handles,
dtype,
container='',
shared_name='',
name=None
)
Входной sparse_handles должен быть int64 матрицей формы [N, 1], где N — размер мини-пакета, а строки соответствуют выходным дескрипторам AddSparseToTensorsMap или AddManySparseToTensorsMap. Ранги исходных объектов SparseTensor, переданных указанным входным операциям, должны совпадать. После создания итогового SparseTensor его ранг на единицу больше рангов входящих объектов SparseTensor (они объединяются вдоль нового измерения строк слева).
Значения формы выходного объекта SparseTensor для всех измерений, кроме первого, равны максимальным значениям формы входных объектов SparseTensor для соответствующих измерений. Первое значение формы — N, размер мини-пакета.
Предполагается, что индексы входных объектов SparseTensor упорядочены в стандартном лексикографическом порядке. Если это не так, после этого шага выполните SparseReorder, чтобы восстановить порядок индексов.
Например, если дескрипторы представляют входные данные — матрицу [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]
| Аргументы | |
|---|---|
sparse_handles | Tensor типа int64. 1-D, сериализованные объекты N SparseTensor. Форма: [N]. |
dtype | tf.DType. dtype объектов SparseTensor, хранящихся в SparseTensorsMap. |
container | Необязательный string. По умолчанию — "". Имя контейнера для SparseTensorsMap, считываемого этой операцией. |
shared_name | Необязательный string. По умолчанию — "". Общее имя для SparseTensorsMap, считываемого этой операцией. Оно не должно быть пустым; следует использовать shared_name или уникальное имя Operation операции, создавшей исходный SparseTensorsMap. |
name | Имя операции (необязательно). |
| Возвращаемые значения | |
|---|---|
Кортеж объектов Tensor (sparse_indices, sparse_values, sparse_shape). | |
sparse_indices | Tensor типа int64. |
sparse_values | Tensor типа dtype. |
sparse_shape | Tensor типа int64. |
© 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/raw_ops/TakeManySparseFromTensorsMap