AVX实现了Python的LogSumExp吗?

2024-05-08 07:18:57 发布

您现在位置:Python中文网/ 问答频道 /正文

我的项目在算法中大量使用logsumexp。目前我正在使用这个库https://github.com/rmcgibbo/logsumexp,它是在SSE指令集中实现的。你知道吗

然而,现代Intel CPU拥有强大的AVX指令集。因此,我想知道是否有更快的logsumexp的AVX实现,甚至是Python的CUDA实现?你知道吗

谢谢你。你知道吗


Tags: 项目httpsgithubcom算法指令cpucuda