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]))