tf.required_space_to_batch_paddings
| Просмотреть исходный код на GitHub |
Вычисление необходимых добавлений для разделения формы блока на форму входных данных.
tf.required_space_to_batch_paddings(
input_shape, block_shape, base_paddings=None, name=None
)
Эта функция может быть использована для расчета подходящего аргумента paddings для использования с space_to_batch_nd и batch_to_space_nd.
| Аргументы | |
|---|---|
input_shape | int32 тензор формы [N]. |
block_shape | int32 тензор формы [N]. |
base_paddings | Необязательный int32 тензор формы [N, 2]. Указывает минимальное количество добавлений для использования. Все элементы должны быть >= 0. Если не указано, по умолчанию используется 0. |
name | строка. Необязательный префикс имени. |
| Возвращаемые значения | |
|---|---|
| (paddings, crops), где:
| |
satisfying | paddings[i, 0] = base_paddings[i, 0]. 0 <= paddings[i, 1] - base_paddings[i, 1] < block_shapei % block_shape[i] == 0 crops[i, 0] = 0 crops[i, 1] = paddings[i, 1] - base_paddings[i, 1] |
Возбуждает: ValueError, если вызов с несовместимыми формами.
© 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/required_space_to_batch_paddings