Tensorflow数据集导入嵌套列表

时间:2018-07-23 11:34:21

标签: tensorflow tensorflow-datasets

我最近正在尝试将列表列表输入Tensorflow tf.data.Dataset

另一个链接How to input a list of lists with different sizes in tf.data.Dataset可以处理诸如[[4,2],[3,4,5]]之类的列表。

但是不适用于嵌套列表,例如[[[4,2],[3,4,5]],[[4,2],[3,4,5]]]。

有没有办法解决这个问题?因为我不想提前填充序列。

我能想到的最好的方法是使用形状为[4,2,*]的嵌套列表(即已知前两个形状),构建2个数据集,每个数据集的输入形状为[4,1,*]。但是,如果第二维也未知,那么这显然将行不通。

0 个答案:

没有答案