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 | A Tensor типа int64. 1-D, сериализованные объекты SparseTensor. Форма: [N]. |
dtype | tf.DType. Тип dtype объектов SparseTensor, хранящихся в SparseTensorsMap. |
container | Необязательный string. По умолчанию "". Имя контейнера для SparseTensorsMap , считанного этим оператором. |
shared_name | Необязательный string. По умолчанию "". Общее имя для SparseTensorsMap , считанного этим оператором. Он не должен быть пустым; вместо этого следует использовать имя shared_name или уникальное имя операции оператора, который создал исходный SparseTensorsMap . |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
Кортеж объектов Tensor (sparse_indices, sparse_values, sparse_shape). | |
sparse_indices | A Tensor типа int64. |
sparse_values | A Tensor типа dtype. |
sparse_shape | A Tensor типа int64. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/raw_ops/TakeManySparseFromTensorsMap