Function get_batch_stats

diff2flow/helpers.py:163–170

(x: Tensor, stats=None)

def get_batch_stats(x: Tensor, stats=None):
    out = {
        "min": torch.min(x).item(),
        "max": torch.max(x).item(),
        "mean": torch.mean(x).item(),
        "std": torch.std(x).item(),
    }
    return out


def un_normalize_ims(ims):