Python 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):
      # ...

如果指定，它必须返回带有给定 dtype 的 Tensor。如果转换函数创建一个新的 Tensor ，它应该使用给定的 name(如果指定)。所有异常都将传播给调用者。

对于某些输入，转换函数可能会返回NotImplemented。在这种情况下，转换过程将继续尝试后续的转换函数。

如果 as_ref 为真，则函数必须返回 Tensor 引用，例如 Variable 。

注意：转换函数将按优先级顺序执行，然后是注册顺序。要确保转换函数 F 在另一个转换函数 G 之前运行，请确保 F 的注册优先级低于 G 。