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