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/versions/r2.9/api_docs/python/tf/raw_ops/SparseSplit