def METHOD_NAME():
    if is_input_lod_tensor:
        total_sequence_length_of_mini_batch = input_lod_data[-1][-1]
        input_data = np.random.randint(
            0,
            10,
            size=(total_sequence_length_of_mini_batch, 1),
            dtype=input_output_data_type)
        return input_data
    else:
        return np.random.randint(
            0,
            10,
            size=input_2d_tensor_shape,
            dtype=input_output_data_type)