tensorflow - 镶木地板中的 tensorflow_io 数组元素失败

Question

test_data = pd.DataFrame({'a':[[1,2,3],[4,5,6]], 'b':['q','p']})
test_data.to_parquet('a.parquet')

columns_init = {"a.list.item": tf.TensorSpec(tf.TensorShape(1), tf.int64), 
                "b": tf.TensorSpec(tf.TensorShape([]), tf.int64)}
another_data = tfio.IODataset.from_parquet('a.parquet').as_numpy_iterator()

another_data.next()

结果是

OrderedDict([(b'a.list.item', 1), (b'b', b'q')])

似乎只加载数组的第一个元素

标签： tensorflow