tf.compat.v1.min_max_variable_partitioner
Функция разбиения для выделения минимального размера каждому срезу.
tf.compat.v1.min_max_variable_partitioner(
max_partitions=1,
axis=0,
min_slice_size=(256 << 10),
bytes_per_string_element=16
)
Возвращает функцию разбиения, которая разбивает переменную заданной формы и типа данных так, чтобы каждый раздел содержал не менее min_slice_size среза переменной. Максимальное количество таких разделов (верхняя граница) задаётся значением max_partitions.
| Аргументы | |
|---|---|
max_partitions | Верхняя граница количества разделов. По умолчанию — 1. |
axis | Ось, вдоль которой разбивается переменная. По умолчанию — 0. |
min_slice_size | Минимальный размер среза переменной для каждого раздела. По умолчанию — 256K. |
bytes_per_string_element | Если Variable имеет тип string, этот параметр задаёт оценку размера каждого скалярного значения в Variable. |
| Возвращает | |
|---|---|
Функцию разбиения, которую можно использовать в качестве аргумента partitioner для variable_scope и get_variable. |
© 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/compat/v1/min_max_variable_partitioner