python - 如何添加嵌套字典作为 tf.data.Dataset.from_tensor_slices 的输入

Question

我正在尝试使用该tf.data.Dataset.from_tensor_slices命令加载数据集。

我的输入是以下格式的嵌套字典列表：

a_dict = {  'a' : 'blablabla',
            'b' : {
                    'c': (tf.constant([[0.390, 0.146]])),
                    'd': (tf.constant([0]))                   
                  }
         }

b_dict = {  'a' : 'blablabla',
            'b' : {
                    'c': (tf.constant([[0.453, 0.655], [0.345, 0.784]])),
                    'd': (tf.constant([0, 0]))                   
                  }
         }        
     

更新： 反序列化输入将不起作用，因为命令：（
将dataset = tf.data.Dataset.from_tensor_slices(pd.DataFrame.from_dict(pd.json_normalize(train_data)).to_dict(orient="list")))给出错误

“所有输入的形状必须匹配：”

有谁知道如何使用 tf.dataset 从具有不同结构的嵌套字典列表中加载数据？

先感谢您！