tf.raw_ops.SparseSplit

Разделить SparseTensor на num_split тензоры по одному измерению.

Просмотр псевдонимов

Псевдонимы для миграции

См. Руководство по миграции для получения более подробной информации.

tf.compat.v1.raw_ops.SparseSplit

tf.raw_ops.SparseSplit(
    split_dim, indices, values, shape, num_split, name=None
)

Если shape[split_dim] не является целым кратным num_split. Сlices [0 : shape[split_dim] % num_split] получает дополнительное измерение. Например, если split_dim = 1 и num_split = 2 и входной данные

input_tensor = shape = [2, 7]
[    a   d e  ]
[b c          ]

Графически выходные тензоры:

output_tensor[0] = shape = [2, 4]
[    a  ]
[b c    ]

output_tensor[1] = shape = [2, 3]
[ d e  ]
[      ]

Аргументы
`split_dim`	A `Tensor` типа `int64`. 0-D. Измерение, по которому следует производить разделение. Должно находиться в диапазоне `[0, rank(shape))`.
`indices`	A `Tensor` типа `int64`. 2-мерный тензор, представляющий индексы разреженного тензора.
`values`	A `Tensor`. 1-мерный тензор, представляющий значения разреженного тензора.
`shape`	A `Tensor` типа `int64`. 1-мерный. тензор, представляющий форму разреженного тензора. output indices: список 1-мерных тензоров, представляющих индексы выходных разреженных тензоров.
`num_split`	An `int`, который является `>= 1`. Количество способов разделения.
`name`	Имя операции (необязательно).

Возвращаемое значение
Кортеж объектов `Tensor` (output_indices, output_values, output_shape).
`output_indices`	Список `num_split` `Tensor` объектов типа `int64`.
`output_values`	Список `num_split` `Tensor` объектов с тем же типом, что и `values`.
`output_shape`	Список `num_split` `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.3/api_docs/python/tf/raw_ops/SparseSplit