# rfft2和fft2的区别: 完整的fft2输出包含冗余信息, rfft2则忽略最后一个维度中的负频率(因此指定进行rfft2的dim会减少) import math import torch import torch.nn as nn import torch.fft as fft # 1. 对空间域(h,w)进行fft B, H, W, D = 1, 32, 32, 256 img_embedding = torch.randn(B, ...