thrust::transform_n

Defined in thrust/transform.h

template<typename InputIterator, typename OutputIterator, typename UnaryFunction>
OutputIterator thrust::transform_n(InputIterator first, ::cuda::std::iter_difference_t<InputIterator> count, OutputIterator result, UnaryFunction op)

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