Exemplos de tensor::device em C++ (Cpp)

Exemplo n.º 1

Arquivo: curand_dlibapi.cpp Projeto: 20337112/dlib

        void curand_generator::
        fill_uniform (
            tensor& data
        )
        {
            if (data.size() == 0)
                return;

            CHECK_CURAND(curandGenerateUniform((curandGenerator_t)handle, data.device(), data.size()));
        }

Exemplo n.º 2

Arquivo: curand_dlibapi.cpp Projeto: 20337112/dlib

        void curand_generator::
        fill_gaussian (
            tensor& data,
            float mean,
            float stddev
        )
        {
            if (data.size() == 0)
                return;

            CHECK_CURAND(curandGenerateNormal((curandGenerator_t)handle, 
                                        data.device(),
                                        data.size(),
                                        mean,
                                        stddev));
        }