Python tf.types.experimental.TensorLike用法及代碼示例

這個符號是一個類型別名。

可以通過 tf.convert_to_tensor 轉換為 tf.Tensor 的所有類型的聯合。

資源：

用法

TensorLike = Union[
    tensorflow.python.types.core.Tensor,
    tensorflow.python.types.core.TensorProtocol,
    int,
    float,
    bool,
    str,
    bytes,
    complex,
    tuple,
    list,
    numpy.ndarray,
    numpy.generic
]

這個定義可以用在用戶代碼中。隨著支持更多輸入類型，將來可能會添加其他類型。

例子：

def foo(x:TensorLike):
  pass

此定義通過靜態類型驗證：

foo(tf.constant([1, 2, 3]))
foo([1, 2, 3])
foo(np.array([1, 2, 3]))