tf.required_space_to_batch_paddings
| Просмотреть исходный код на GitHub |
Вычислить необходимые отступы для того, чтобы размер блока block_shape делил размер входного массива input_shape.
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, если вызывается с несовместимыми формами.
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/required_space_to_batch_paddings