tensorflow的dataset.map函数

问题发生背景

tensorflow的数据预处理函数

问题相关代码,请勿粘贴截图

def pre_compute(example):
example['descript_reshape'], example['descrpt_deriv'], example['rij'], example['nlist'], example['dim'] = self.model.build_R (example['coord'],
example['type'],
example['natoms_vec'],
example['box'],
example['default_mesh'],
suffix = "",
reuse = False)
return example
def build_iter(self, data_dict):
dataset = tf.data.Dataset.from_tensor_slices(data_dict)#是一个字典,value是numpy二维数组
dataset = dataset.map(map_func=self.pre_compute, num_parallel_calls=tf.data.experimental.AUTOTUNE)
dataset = dataset.batch(1)
dataset = dataset.prefetch(batch_size=1)
iter = dataset.make_one_shot_iterator()
next_element = iter.get_next()

我想知道如果我Sess.run(next_element)得到的是不是一个字典