将complex64数据输入Keras顺序模型

时间:2019-03-28 13:20:34

标签: python keras sequential

我正在训练傅立叶域的CNN。为了加快训练速度,我想到了在训练之前先对整个数据集进行fft并将该数据馈送到顺序模型中。但是在模型的第一层(即自定义Keras层)内部,训练数据显示为具有float32数据类型。顺序模型是否仅吸收实际输入数据? 谢谢。

1 个答案:

答案 0 :(得分:0)

添加一个InputLayer(... dtype ='complex64')层,即,将数据类型指定为'complex64'的InputLayer()作为顺序模型的第一层,使我可以将complex64数据传递给该模型。 / p>

相关问题