emd for pytorch code ?

[iganggang]

could it work in CUDA? I want to make it as a layer in PatchTST(a time series forecasting model) class emd(): def init(self, ): super(emd, self).init() self.emd = emd() def forward(self, x): imf1, imf2 = emd(x) return imf1, imf2