def METHOD_NAME(function, elements):
    if len(elements) == 1:
        return function(elements)
    else:
        batch_size = elements[0].shape[0]
        output_store = list()
        for index in range(batch_size):
            output_store.append(function([x[index] for x in elements]))
        return np.stack(output_store)