Aurelius84

  • CUDA 的随机数算法 API

    参考自 Nvidia cuRand 官方 API 文档 一、具体使用场景 如下是是在 dropout 优化中手写的 uniform_random 的 Kernel: #include <cuda_runtime.h> #include <curand_kernel.h> __device__ inline float cinn_nvg…

    人工智能概论 2023年5月6日
    00
合作推广
合作推广
分享本页
返回顶部