tf.keras.ops.get_item

Return x[key].