我正在尝试编写一个输入管道,该管道将接收一个信号数据集(每个条目代表一个帧信号)和一个噪声数据集(每个条目代表一个帧噪声)并将它们混合在一起成为一批帧。
混合很简单:
output = signal[signal_frame_start:signal_frame_start+n] + noise[noise_frame_start:noise_frame_start+n]*gain
每个信号(和噪声)都通过tf.data.TFRecordDataset
访问。我的主要问题是我不知道如何编写有效的管道来执行此操作。
我尝试编写生成器函数,但无法使其并行化。
任何有关如何做这种事情的想法将不胜感激。我也希望在多个内核上运行它(16)。