TensorFlow函數(shù)教程：tf.lite.OpHint.OpHintArgumentTracker

tf.lite.OpHint.OpHintArgumentTracker函數(shù)

類 OpHintArgumentTracker

別名：

• 類 tf.contrib.lite.OpHint.OpHintArgumentTracker
• 類 tf.lite.OpHint.OpHintArgumentTracker

定義在：tensorflow/lite/python/op_hint.py。

從概念上跟蹤“OpHint函數(shù)”的參數(shù)索引。

這些函數(shù)的輸入和參數(shù)都使用類的實例，因此它們可以具有獨立的編號。

__init__(
    function_name,
    unique_function_id,
    node_name_prefix,
    attr_name
)

初始化ophint參數(shù)。

參數(shù)：

• function_name：跟蹤參數(shù)的函數(shù)的名稱。
• unique_function_id：跟蹤參數(shù)的函數(shù)的UUID。
• node_name_prefix：如何命名創(chuàng)建的標(biāo)識。
• attr_name：用于存儲此提示的索引的屬性的名稱。即FUNCTION_INPUT_INDEX或FUNCTION_OUTPUT_INDEX

方法

add(
    arg,
    tag=None,
    name=None,
    aggregate=None,
    index_override=None
)

返回輸入張量的包裝張量作為參數(shù)。

參數(shù)：

• arg：TensorFlow張量應(yīng)該被認(rèn)為是一個參數(shù)。
• tag：String標(biāo)記，用于標(biāo)識應(yīng)打包的參數(shù)。
• name：參數(shù)名稱。這包含在Identity提示操作名稱中。
• aggregate：聚合戰(zhàn)略?？山邮艿闹禐镺pHint.AGGREGATE_FIRST，OpHint.AGGREGATE_LAST和OpHint.AGGREGATE_STACK。注意，聚合僅在指定標(biāo)記時有效。
• index_override：指定最終存根中的輸入/輸出索引。即，add（arg0，index = 1）; add（arg1，index = 0）將使最終存根作為stub_func(inputs[arg1, arg0], outputs=[])，而不是基于默認(rèn)的調(diào)用順序的排序。

返回：

表示包裝參數(shù)的張量。

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

• ValueError：當(dāng)指數(shù)不一致時。