import torch from torch.autograd import Function ... from numpy.fft import rfft2, irfft2 class BadFFTFunction(Function): @staticmethod def forward(ctx, ...
確定! 回上一頁