我正在尝试创建一个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
,我该怎么做?
或者我还有其他方法可以做到吗?
这是新手,谢谢您的帮助!
答案 0 :(得分: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))