tf.sparse.split
| Просмотреть исходный код на GitHub |
Разделение SparseTensor на num_split тензоров вдоль axis. (устаревшие аргументы)
tf.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 для * (временный для изменения порядка аргументов) |
sp_input | Разделяемый SparseTensor . |
num_split | Целое число Python. Количество способов разделения. |
axis | 0-мерный int32 Tensor. Измерение, по которому следует производить разделение. |
name | Имя операции (необязательно). |
split_dim | Устаревшее старое имя для axis. |
| Возвращаемое значение | |
|---|---|
num_split SparseTensor объекты, полученные в результате разделения value. |
| Возбуждает | |
|---|---|
TypeError | Если sp_input не является SparseTensor . |
ValueError | Если устаревшие split_dim и axis оба не None. |
© 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/sparse/split