لایه lambda توی کراس رو تست کردید؟
برای این لایه باید یه تابع خودتون بنوسید که یه ورودی بگیره و یه خروجی بده
که الان تابع برای کاربرد شما میشه همین که تبدیل فرویه یا معکوسش رو حساب کنه
تا حالا خودم تست نکردم اما به نظرم اگه مشکلی با نوع داده که complex ه نباشه، با همون Lambda بشه حلش کرد