跨批次维度的TF RaggedTensor映射

Question

我有一个文本分类任务，其中每个训练示例都是RaggedTensor。现在，我的培训批次是RaggedTensor本身（具有1个额外的维度）。我需要以某种方式能够说出在训练期间批次大小（批次大小）是固定的，因此我可以在每个示例中映射一个操作。 splitRDD.foreach(x => println(x.mkString(", ")) 不能使所有内容变平，tf.ragged.map_flat_values会产生帮助：

tf.map_fn

请帮助。