Function SetTensorToRandomValue

tests/common/TensorDataUtils.hpp:451–487

template<typename DT>
static void SetTensorToRandomValue(const TensorData &tensorData, DT minVal, DT maxVal, int sample)
{
    if (!nvcv::TensorDataAccessStrided::IsCompatible(tensorData))
        throw std::runtime_error("Tensor Data is not pitch access capable.");

    auto tDataAc = nvcv::TensorDataAccessStrided::Create(tensorData);

    if (tDataAc->numSamples() <= sample)
        throw std::runtime_error("Number of samples smaller than requested sample.");

    int                        inElements = (tDataAc->sampleStride() / sizeof(DT));
    std::vector<DT>            srcVec(inElements);
    std::default_random_engine randEng(0);

    int totalSamples;
    if (sample < 0)
    {
        totalSamples = tDataAc->numSamples();
        sample       = 0;
    }
    else
    {
        totalSamples = sample + 1;
    }
    std::uniform_int_distribution<DT> srcRand{minVal, maxVal};
    for (int i = sample; i < totalSamples; ++i)
    {
        std::generate(srcVec.begin(), srcVec.end(), [&]() { return srcRand(randEng); });
        if (cudaSuccess
            != cudaMemcpy(tDataAc->sampleData(i), srcVec.data(), tDataAc->sampleStride(), cudaMemcpyHostToDevice))
        {
            throw std::runtime_error("CudaMemcpy failed");
        }
    }

    return;
}

template<typename DT>
void SetTensorFromVector(const TensorData &tensorData, std::vector<DT> &data, int sample)