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 。