Как мне вставить два предложения в одну форму с помощью tf.data.Dataset.padded_batch - PullRequest
1 голос
/ 19 марта 2019

Я хочу построить модель seq2seq и создать набор данных tenorflow с парой {source, target}.Когда я попытался дополнить эти два вектора, я обнаружил, что не могу добавить их в одну и ту же форму

batched_train = dataset_train.padded_batch(32,padded_shapes={
   "source":tf.TensorShape([None]),
   "target":tf.TensorShape([None])}, drop_remainder=True)
batched_val = dataset_val.padded_batch(32,padded_shapes={
   "source":tf.TensorShape([None]),
   "target":tf.TensorShape([None])}, drop_remainder=True)

. И я получаю форму источника (32, 107), а форма цели (32, 180), я брожу, как я могу это исправить.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...