thrust::transform_if_n

Defined in thrust/transform.h

template<typename InputIterator1, typename InputIterator2, typename ForwardIterator, typename UnaryFunction, typename Predicate>
ForwardIterator thrust::transform_if_n(InputIterator1 first, ::cuda::std::iter_difference_t<InputIterator1> count, InputIterator2 stencil, ForwardIterator result, UnaryFunction op, Predicate pred)

Like transform_if, but uses an element count instead of an iterator to the last element of the input sequence.