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 | Минимальный размер среза переменной на раздел. По умолчанию 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/r2.3/api_docs/python/tf/compat/v1/min_max_variable_partitioner