def unique_idx(x):
    output = tf.unique(x)
    return output[1]

inp1 = Input(batch_shape(None, 1))
idx = Lambda(unique_idx)(inp1)

model = Model(inputs=inp1, outputs=idx)