Spec-Zone.ru › TensorFlow

tf.register_tensor_conversion_function

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

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

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

Подробнее см. в руководстве по миграции.

tf.compat.v1.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 Если аргументы имеют неподходящий тип.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/register_tensor_conversion_function

Spec-Zone.ru

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