

def gather(reference, indices):
    'Retrieves the vectors of indices `indices`\n    in the 2D tensor `reference`.\n\n    # Arguments\n        reference: a 2D tensor.\n        indices: an int tensor of indices.\n\n    # Returns\n        A 3D tensor of same type as `reference`.\n    '
    return tf.gather(reference, indices)

dataset/ETHPy150Open fchollet/keras/keras/backend/tensorflow_backend.py gather/original