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.