mindflow.loss.WaveletTransformLoss

.. 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。