Подать данные complex64 в последовательную модель Keras

Question

Я работаю над обучением CNN в области Фурье.Чтобы ускорить тренировку, я подумал о том, чтобы взять fft всего набора данных перед тренировкой и передать эти данные в последовательную модель.Но внутри первого слоя модели, который является пользовательским слоем Keras, показано, что обучающие данные имеют тип данных float32.Последовательная модель принимает только реальные входные данные?Спасибо.

Answer 1

Добавление слоя InputLayer (... dtype = 'complex64'), то есть InputLayer () с типом данных, указанным как 'complex64' в качестве первого слоя последовательной модели, позволил мне передавать данные complex64 в модель.