Python tf.experimental.ExtensionType用法及代码示例

TensorFlow ExtensionType 类的基类。

用法

tf.experimental.ExtensionType(
    *args, **kwargs
)

Tensorflow ExtensionType 类是专门的 Python 类，可以透明地与 TensorFlow 一起使用——例如，它们可以与 tf.cond 或 tf.while_loop 等操作一起使用，并用作 tf.function 和 Keras 层的输入或输出。

新的ExtensionType 类是通过创建包含所有实例变量的类型注释的tf.ExtensionType 的子类来定义的。支持以下类型注释：

类型	示例
Python 整数	i:int
Python 浮点数	f:float
Python 字符串	s:str
Python 布尔值	b:bool
Python 无	n:None
Tensors	t:tf.Tensor
复合张量	rt:tf.RaggdTensor
扩展类型	m:MyMaskedTensor
张量形状	shape:tf.TensorShape
张量类型	dtype:tf.DType
类型联合	length:typing.Union[int, float]
Tuples	params:typing.Tuple[int, float, int, int]
带省略号的元组	lengths:typing.Tuple[int, ...]
Mappings	tags:typing.Mapping[str, str]

使用 typing.Mapping 注释的字段将使用不可变映射类型进行存储。

ExtensionType 值是不可变的——即，一旦构造，您就不能修改或删除它们的任何实例成员。

例子

class MaskedTensor(ExtensionType):
  values:tf.Tensor
  mask:tf.Tensor

class Toy(ExtensionType):
  name:str
  price:ops.Tensor
  features:typing.Mapping[str, tf.Tensor]

class ToyStore(ExtensionType):
  name:str
  toys:typing.Tuple[Toy, ...]