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`	Целое число, равное `>= 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