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 | Базовый тип или кортеж базовых типов для всех объектов, которые 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