tf.min_max_variable_partitioner
Разделитель для выделения минимального размера каждого фрагмента.
tf.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 | Минимальный размер фрагмента переменной в каждом разделе. По умолчанию 256 КБ. |
bytes_per_string_element | Если тип Variable — строка, это обеспечивает оценку размера каждого скаляра в Variable. |
| Возвращаемое значение | |
|---|---|
Функция разбиения, которая может быть использована в качестве аргумента partitioner для variable_scope и get_variable. |
© 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/r1.15/api_docs/python/tf/min_max_variable_partitioner