tf.create_partitioned_variables
Создайте список разделяемых переменных в соответствии с заданным slicing. (устарело)
tf.create_partitioned_variables(
shape, slicing, initializer, dtype=tf.dtypes.float32, trainable=True,
collections=None, name=None, reuse=None
)
В настоящее время можно разрезать только одну размерность полной переменной, а полную переменную можно восстановить путем конкатенации возвращаемого списка по этой размерности.
| Аргументы | |
|---|---|
shape | Список целых чисел. Форма полной переменной. |
slicing | Список целых чисел. Как разделить переменную. Должен иметь такую же длину, как shape. Каждое значение указывает, сколько фрагментов нужно создать в соответствующей размерности. В настоящее время только одно из значений может быть больше 1; то есть, переменная может быть разделена только по одной размерности. Для удобства, не обязательно, чтобы запрошенное количество разделов делило соответствующую размерность без остатка. Если это не так, размеры фрагментов увеличиваются на 1, начиная с раздела 0, пока весь избыток не будет поглощен. Правила корректировки могут измениться в будущем, но поскольку вы можете сохранять/восстанавливать эти переменные с различными параметрами разбиения, это не должно представлять проблемы. |
initializer | A Tensor формы shape или функция инициализации переменной. Если функция, она будет вызвана один раз для каждого фрагмента, передав форму и тип данных фрагмента в качестве параметров. Функция должна вернуть тензор с такой же формой, как у фрагмента. |
dtype | Тип переменных. Игнорируется, если initializer является Tensor. |
trainable | Если True, также добавьте все переменные в коллекцию графа GraphKeys.TRAINABLE_VARIABLES. |
collections | Список ключей коллекций графа, в которые нужно добавить переменные. По умолчанию [GraphKeys.GLOBAL_VARIABLES]. |
name | Необязательное имя для полной переменной. По умолчанию "PartitionedVariable" и автоматически получает уникальное имя. |
reuse | Булево или None; если True и имя установлено, будут повторно использованы ранее созданные переменные. Если False, будут созданы новые переменные. Если None, будет унаследовано повторное использование родительского области. |
| Возвращаемые значения | |
|---|---|
| Список переменных, соответствующих разбиению. |
| Исключения | |
|---|---|
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/r1.15/api_docs/python/tf/create_partitioned_variables
