再求教一个问题：为什么WaveletsTransform传入的张量要设置成五维的？

[qi-jinnian]

在你的utils文件中的WaveletsTransform函数中wavelets = self.transform(x=x.unsqueeze(1))，为什么要对输入的x进行unsqueeze(1),使它变成五维的张量，这样对于二维小波变换不是错误的吗？

[brunobelloni]

its because this library i've been using requires another dimension to apply wavelet transform.

from pytorch_wavelets import DWTInverse, DWTForward

so i apply the unsqueeze on library's input and squeeze on library's output.