tf.sparse.reset_shape
| Просмотреть исходный код на GitHub |
Переустанавливает форму 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.
| Аргументы | |
|---|---|
sp_input | Входной SparseTensor. |
new_shape | None или вектор, представляющий новую форму возвращаемого SparseTensor. |
| Возвращаемое значение | |
|---|---|
SparseTensor с неизменными индексами и значениями от sp_input. Его форма — new_shape если она задана. В противном случае — наименьшая ограничивающая рамка sp_input |
| Исключения | |
|---|---|
TypeError | Если sp_input не является SparseTensor. |
ValueError | Если new_shape представляет тензор с другим рангом, чем у sp_input (если формы известны при построении графа). |
ValueError | Если new_shape определено во время построения графа, как имеющее размеры измерений слишком маленькие. |
OpError |
|
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/sparse/reset_shape