如何将元组列表（tf.constant，tf.constant）转换为张量流数据集？

Question

我正在尝试创建一个 Transformer 模型，我有 2 个np.arrays，都有字符串，我用它们来创建一个元组列表

元组的格式是：

class 'tuple' (tf.Tensor: shape=(), dtype=string, numpy=b'abc', tf.Tensor: shape=(), dtype=string, numpy=b'xyz')

我想将这些元组组合成tensorflow.python.data.ops.dataset_ops._OptionsDataset，我该怎么做？

或者有没有其他方法可以做到？

新手，感谢您的帮助！

Answer 1

您可以使用tensor_from_slices。 Here.

编辑

Example
# Two tensors can be combined into one Dataset object. 
values1 = tf.constant(['A', 'B', 'A']) # ==> 3x1 tensor
values2 = tf.constant(['A', 'B', 'A']) # ==> 3x1 tensor
dataset = Dataset.from_tensor_slices((values1, values2))