Function generate_torch_cuda_randn

src/framework/sampling/torch_random.cpp:177–185

        }
        output[index] = value;
    }
}

std::vector<float> generate_torch_cuda_randn(
    size_t count,
    uint64_t seed,
    TorchRandnPrecision precision,
    uint64_t start_index) {
    std::vector<float> output(count);
    fill_torch_cuda_randn(output.data(), output.size(), seed, precision, start_index);
    return output;
}