用于组合两个Tensorflow数据集的输入管道

时间:2019-12-14 23:34:44

标签: python tensorflow tensorflow-datasets tensorflow2.0

我正在尝试编写一个输入管道,该管道将接收一个信号数据集(每个条目代表一个帧信号)和一个噪声数据集(每个条目代表一个帧噪声)并将它们混合在一起成为一批帧。

混合很简单:

output = signal[signal_frame_start:signal_frame_start+n] + noise[noise_frame_start:noise_frame_start+n]*gain

每个信号(和噪声)都通过tf.data.TFRecordDataset访问。我的主要问题是我不知道如何编写有效的管道来执行此操作。

我尝试编写生成器函数,但无法使其并行化。

任何有关如何做这种事情的想法将不胜感激。我也希望在多个内核上运行它(16)。

0 个答案:

没有答案