tf.sparse.split

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

tf.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  ]
[      ]