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, ...]