tf.register_tensor_conversion_function
| Просмотреть исходный код на GitHub |
Регистрирует функцию для преобразования объектов типа 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 | Базовый тип или кортеж базoвых типов для всех объектов, которые 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/versions/r2.9/api_docs/python/tf/register_tensor_conversion_function