def METHOD_NAME(test_case):
    kMaxInputCount = 128 + 1
    stack_list = [
        random_tensor(ndim=2, dim0=3, dim1=4) for _ in range(kMaxInputCount)
    ]
    out = torch.stack(stack_list, 0)
    return out