mindflow.loss.WaveletTransformLoss ================================== .. image:: https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/website-images/master/resource/_static/logo_source.svg :target: https://gitee.com/mindspore/mindscience/blob/master/docs/api_python/mindflow/loss/mindflow.loss.WaveletTransformLoss.rst :alt: 查看源文件 .. py:class:: mindflow.loss.WaveletTransformLoss(wave_level=2, regroup=False) 多级小波变换损失。 参数: - **wave_level** (int) - 小波变换级数,应为正整数。默认值: ``2``。 - **regroup** (bool) - 小波变换损失的regroup误差组合形式。默认值: ``False``。 输入: - **input** - 张量构成的tuple。Tensor的shape为 :math:`(B*H*W/(P*P), P*P*C)` ,其中B表示批次大小。H、W分别表示图像的高度和宽度。P表示补丁大小。C表示特征通道。 输出: Tensor。小波变换损失函数输出。 异常: - **TypeError** - 如果 `wave_level` 不是int。 - **TypeError** - 如果 `regroup` bool。