Spec-Zone.ru › TensorFlow 1.15

tf.register_tensor_conversion_function

Просмотр исходного кода на GitHub

Регистрирует функцию для преобразования объектов типа base_type в Tensor.

Просмотр псевдонимов

Псевдонимы для миграции

См. руководство по миграции для получения более подробной информации.

tf.compat.v1.register_tensor_conversion_function, `tf.compat.v2.register_tensor_conversion_function`

tf.register_tensor_conversion_function(
    base_type, conversion_func, priority=100
)

Функция преобразования должна иметь следующий вид:

def conversion_func(value, dtype=None, name=None, as_ref=False):
  # ...

Она должна возвращать Tensor с заданным dtype (если указано). Если функция преобразования создает новый Tensor, она должна использовать заданный name (если указано). Все исключения будут переданы вызывающей стороне.

Функция преобразования может возвращать NotImplemented для некоторых входных данных. В этом случае процесс преобразования продолжит попытку последующих функций преобразования.

Если as_ref равно true, функция должна возвращать ссылку на Tensor, например, Variable.

Примечание: Функции преобразования будут выполняться в порядке приоритета, а затем в порядке регистрации. Чтобы убедиться, что функция преобразования F выполняется перед другой функцией преобразования G, необходимо, чтобы F была зарегистрирована с меньшим приоритетом, чем G.
Аргументы
base_type Базовый тип или кортеж базовых типов для всех объектов, которые conversion_func принимает.
conversion_func Функция, которая преобразует экземпляры base_type в Tensor.
priority Необязательное целое число, которое указывает приоритет применения этой функции преобразования. Функции преобразования с меньшими значениями приоритета выполняются раньше, чем функции преобразования с большими значениями приоритета. По умолчанию равно 100.
Возможные исключения
TypeError Если аргументы не имеют соответствующего типа.

© 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/register_tensor_conversion_function

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API