将complex64数据输入Keras序列模型

Question

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

Answer 1

添加InputLayer（... dtype ='complex64'）图层，即具有指定为'complex64'的数据类型的InputLayer（）作为顺序模型的第一层，允许我将complex64数据传递给模型。