tf.required_space_to_batch_paddings
Вычисляет заполнение, необходимое для того, чтобы 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 | string. Необязательный префикс имени. |
| Возвращает | |
|---|---|
| (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/api_docs/python/tf/required_space_to_batch_paddings