tf.compat.v1.create_partitioned_variables
Создать список разделяемых переменных в соответствии с заданным slicing. (устарело)
tf.compat.v1.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/r2.3/api_docs/python/tf/compat/v1/create_partitioned_variables