tf.sparse.reshape

Преобразует SparseTensor для представления значений в новой плотной форме.

tf.sparse.reshape(
    sp_input, shape, name=None
)

Эта операция имеет те же семантику, что и reshape на представленном плотном тензоре. Индексы ненулевых значений в sp_input пересчитываются на основе новой плотной формы, и возвращается новый SparseTensor, содержащий новые индексы и новую форму. Порядок ненулевых значений в sp_input не изменяется.

Если один компонент shape имеет специальное значение -1, размер этого измерения вычисляется таким образом, чтобы общий плотный размер оставался постоянным. Максимум один компонент в shape может быть равен -1. Количество плотных элементов, подразумеваемое shape, должно совпадать с количеством плотных элементов, первоначально представленных sp_input.

Например, если sp_input имеет форму [2, 3, 6] и indices / values:

[0, 0, 0]: a
[0, 0, 1]: b
[0, 1, 0]: c
[1, 0, 0]: d
[1, 2, 3]: e

и shape равно [9, -1], то результатом будет SparseTensor формы [9, 4] и indices / values:

[0, 0]: a
[0, 1]: b
[1, 2]: c
[4, 2]: d
[8, 1]: e