def METHOD_NAME(array, indices, axis):
    """Take the value from the input array on the given indices along the
    given axis.
    # Arguments
        array: Array
        indices: list/Array. values to be gathered from
        axis: Int. Axis along which to gather values.
    # Returns
        Array. Gathered values from the input array
    """
    gathered = np.take_along_axis(array, indices, axis=axis)
    return gathered