tf.compat.v1.sparse_split
Разделяет SparseTensor на num_split тензоров вдоль axis. (устаревшие аргументы)
tf.compat.v1.sparse_split(
keyword_required=KeywordRequired(),
sp_input=None,
num_split=None,
axis=None,
name=None,
split_dim=None
)
Если sp_input.dense_shape[axis] не кратно num_split, каждый срез, начиная с 0:shape[axis] % num_split, получает дополнительный элемент измерения. Например, если axis = 1 и num_split = 2, а входные данные имеют вид:
input_tensor = shape = [2, 7] [ a d e ] [b c ]
Графически выходные тензоры выглядят так:
output_tensor[0] = [ a ] [b c ] output_tensor[1] = [ d e ] [ ]
| Аргументы | |
|---|---|
keyword_required | Заменитель * для Python 2 (временный, для изменения порядка аргументов) |
sp_input | SparseTensor для разделения. |
num_split | Целое число Python. Количество частей, на которые нужно разделить. |
axis | Скалярный int32 Tensor. Измерение, вдоль которого выполняется разделение. Должно находиться в диапазоне [-rank, rank), где rank — количество измерений во входном SparseTensor. |
name | Имя операции (необязательно). |
split_dim | Устаревшее старое имя для axis. |
| Возвращает | |
|---|---|
Объекты num_split SparseTensor, полученные в результате разделения value. |
| Вызывает исключения | |
|---|---|
TypeError | Если sp_input не является SparseTensor. |
ValueError | Если устаревшие split_dim и axis оба не равны None. |
© 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/compat/v1/sparse_split