TensorFlow函數(shù)：將base_type的對象轉(zhuǎn)換為Tensor

tf.register_tensor_conversion_function 函數(shù)

register_tensor_conversion_function(
    base_type,
    conversion_func,
    priority=100
)

定義在：tensorflow/python/framework/ops.py.

參見指南：構(gòu)建圖表>用于構(gòu)建TensorFlow的庫

注冊一個函數(shù),用于將 base_type 的對象轉(zhuǎn)換為 Tensor.

這個轉(zhuǎn)換函數(shù)必須具有以下簽名：

def conversion_func(value, dtype=None, name=None, as_ref=False):
  # ...

如果指定,它必須返回具有給定 dtype 的 Tensor.如果轉(zhuǎn)換函數(shù)創(chuàng)建一個新的 Tensor,它應(yīng)該使用給定的 name(如果指定).所有異常將被傳播給調(diào)用方.

轉(zhuǎn)換函數(shù)可能會為一些輸入返回 NotImplemented.在這種情況下,轉(zhuǎn)換過程將繼續(xù)嘗試后續(xù)的轉(zhuǎn)換函數(shù).

如果 as_ref 為 true,則該函數(shù)必須返回一個Tensor引用,如 Variable.

注意：轉(zhuǎn)換函數(shù)將按照優(yōu)先級順序執(zhí)行,然后是注冊順序.要確保轉(zhuǎn)換函數(shù) F 在另一個轉(zhuǎn)換函數(shù) G 之前運行,請確保函數(shù) F 使用比函數(shù) G 小的優(yōu)先級注冊.

參數(shù)：

• base_type：conversion_func 接受的所有對象的基本類型或基本類型的元組.
• conversion_func：將 base_type 的實例轉(zhuǎn)換為 Tensor 的函數(shù).
• priority：表示應(yīng)用此轉(zhuǎn)換函數(shù)的優(yōu)先級的可選整數(shù).具有較小優(yōu)先級值的轉(zhuǎn)換函數(shù)比具有較大優(yōu)先級值的轉(zhuǎn)換函數(shù)更早運行.默認(rèn)為100.

可能引發(fā)的異常：

• TypeError：如果參數(shù)沒有適當(dāng)?shù)念愋?