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 или уникальное имя операции Op, которая создала исходный 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/versions/r2.9/api_docs/python/tf/raw_ops/TakeManySparseFromTensorsMap