Как преобразовать список кортежей (tf.constant, tf.constant) в набор данных tenorflow? - PullRequest
0 голосов
/ 28 января 2020

Я пытаюсь создать модель 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 голосов
/ 28 января 2020

Вы можете использовать tensor_from_slices. Здесь .

РЕДАКТИРОВАТЬ

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