Function data_from_keys

tests/transforms/compose/test_compose.py:31–38

(keys, h, w)

def data_from_keys(keys, h, w):
    if keys is None:
        data = torch.arange(h * w).reshape(1, h, w)
    else:
        data = {}
        for i_k, k in enumerate(keys):
            data[k] = torch.arange(h * w).reshape(1, h, w).mul_(i_k * h * w)
    return data


class _RandXform(Randomizable):