tf.sparse.reset_shape

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

tf.sparse.reset_shape(
    sp_input, new_shape=None
)

Если new_shape равно None, возвращает копию sp_input с формой, сброшенной до наименьшей ограничивающей рамки sp_input. Это будет форма, состоящая из нулей, если у sp_input нет значений.

Если new_shape задано, оно должно быть не меньше формы sp_input. Когда это условие выполняется, возвращаемый SparseTensor будет иметь форму, сброшенную до new_shape, и его индексы и значения не изменятся по сравнению с sp_input..

Пример:

Рассмотрим sp_input с формой [2, 3, 5]:

• Ошибка установить new_shape как [3, 7], так как это представляет тензор ранга 2, а sp_input имеет ранг 3. Это либо ошибка ValueError во время построения графа (если обе формы известны), либо ошибка OpError во время выполнения.

• Установка new_shape как [2, 3, 6] допустима, так как эта форма не меньше по всем измерениям, чем исходная форма [2, 3, 5].

• С другой стороны, установка new_shape как [2, 3, 4] также является ошибкой: третье измерение меньше исходной формы 2, 3, 5.

• Если new_shape равно None, возвращаемый SparseTensor будет иметь форму [2, 3, 4], которая является наименьшей ограничивающей рамкой sp_input.