tf.register_tensor_conversion_function
Регистрирует функцию для преобразования объектов типа base_type в Tensor.
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