Spec-Zone.ru › TensorFlow 2.3

tf.register_tensor_conversion_function

Просмотреть исходный код на GitHub

Регистрирует функцию для преобразования объектов типа 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 истинно, функция должна возвращать ссылку на 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/r2.3/api_docs/python/tf/register_tensor_conversion_function

Spec-Zone.ru

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