将 complex64 数据输入 Keras 顺序模型

Question

我正在训练傅立叶域中的 CNN。为了加快训练速度，我想到了在训练之前获取整个数据集的 fft，并将这些数据提供给顺序模型。但在模型的第一层（自定义 Keras 层）内，训练数据显示为 float32 数据类型。顺序模型是否只接受真实的输入数据？ 谢谢。

Answer 1

添加一个 InputLayer(... dtype='complex64') 层，即数据类型指定为 'complex64' 的 InputLayer() 作为顺序模型的第一层允许我通过 complex64数据到模型。