如何将元组列表(tf.constant,tf.constant)转换为张量流数据集?

时间:2020-01-28 10:01:42

标签: python numpy tensorflow dataset

我正在尝试创建一个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,我该怎么做?

或者我还有其他方法可以做到吗?

这是新手,谢谢您的帮助!

1 个答案:

答案 0 :(得分:1)

您可以使用tensor_from_slicesHere

编辑

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