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.