thrust::transform_n

Defined in thrust/transform.h

template<typename DerivedPolicy, typename InputIterator, typename OutputIterator, typename UnaryFunction>
OutputIterator thrust::transform_n(const detail::execution_policy_base<DerivedPolicy> &exec, 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.