About 1D FFT AND 2D FFT

[Aerikervalid]

您好，感谢您的工作。我有两个问题想咨询。
1.我在有关序列任务上尝试了你的模型，我的序列长度为256，tensor为（B，256，512）输入到你的模型，我这里忽略了输入的意义，直接将序列256看作是2维信号 (即 H W 为16) ，强行让模型学习，但是奇怪的是结果却很好。我想问一下如何直接使用1D FFT对序列进行操作，代码如何编写？
2.在进行2D FFT，输入的信号必须像文中的图片那样 H W是一样的吗？如何可以不一样，代码需要做那些修改？
打扰了，期待您的回复！再次感谢！