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. Срезы [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`	`Tensor` типа `int64`. 0-мерный тензор. Измерение, вдоль которого выполняется разделение. Должно находиться в диапазоне `[0, rank(shape))`.
`indices`	`Tensor` типа `int64`. 2-мерный тензор, представляющий индексы разреженного тензора.
`values`	`Tensor`. 1-мерный тензор, представляющий значения разреженного тензора.
`shape`	`Tensor` типа `int64`. 1-мерный тензор, представляющий форму разреженного тензора. Выходные индексы: список 1-мерных тензоров, представляющих индексы выходных разреженных тензоров.
`num_split`	`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`.

© 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/SparseSplit