Function torch_cuda_randn_element

src/framework/sampling/torch_random.cpp:74–86

    normal0 = radius * std::sin(angle);
    normal1 = radius * std::cos(angle);
}

float torch_cuda_randn_element(uint64_t seed, uint64_t index) {
    const Philox4 counter{
        0U,
        0U,
        static_cast<uint32_t>(index),
        static_cast<uint32_t>(index >> 32U),
    };
    const Philox4 random = philox_4x32_10(counter, seed);
    float normal0 = 0.0F;
    float normal1 = 0.0F;
    box_muller(random.x, random.y, normal0, normal1);
    return normal0;
}