tf.compat.v2.sparse.split

Разделение SparseTensor на num_split тензоров вдоль axis.

tf.compat.v2.sparse.split(
    sp_input=None, num_split=None, axis=None, name=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  ]
[      ]

Аргументы
`sp_input`	Разделяемый `SparseTensor` .
`num_split`	Целое число Python. Количество способов разделения.
`axis`	0-мерный `int32` `Tensor`. Измерение, вдоль которого происходит разделение.
`name`	Имя операции (необязательно).

Возвращаемые значения
`num_split` `SparseTensor` объектов, полученных в результате разделения `value`.

Исключения
`TypeError`	Если `sp_input` не является `SparseTensor`.

© 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/r1.15/api_docs/python/tf/compat/v2/sparse/split