def METHOD_NAME(self, sizes, N, dim, device):
    random.seed(42)
    inputs = []
    gen_sizes = []
    if type(sizes) == list and N == -1:
        gen_sizes = sizes
    else:
        for i in range(N):
            gen_sizes.append(
                [
                    old_size() if callable(old_size) else old_size
                    for old_size in sizes
                ]
            )
    for s in gen_sizes:
        inputs.append(torch.rand(s, device=device))
    result = torch.empty(0, device=device)
    self.inputs = {"result": result, "inputs": inputs, "dim": dim}
    self.set_module_name("cat")