def METHOD_NAME(i):
    inputs["repeat_times_tensor"] = [
        "repeat_times_tensor_data" + str(i)
        for i in range(repeat_times_len)
    ]
    if (choose_repeat == "repeat_times_tensor"):
        return np.array([repeat_times_data[i]]).astype(np.int32)
    else:
        return np.random.randint(1, 5, []).astype(np.int32)