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.