8个重大版本发布
0.11.0 | 2024年7月5日 |
---|---|
0.9.0 | 2024年5月16日 |
0.6.0 | 2024年3月31日 |
0.1.0 | 2023年12月31日 |
在 机器学习 中排名第 458
每月下载量 903 次
在 6 个crate中使用(通过 rten)
93KB
2K SLoC
rten-vecmath
此crate提供跨平台的SIMD类型,该类型抽象了不同架构上的SIMD内嵌函数。与 std::simd
不同,此功能在稳定的Rust上工作。该crate还包括在运行时检测可用指令并调度到最佳实现的功能。
此crate还包含了一些机器学习模型中性能关键数学函数(如exp、erf、tanh、softmax等)的SIMD向量化版本。
lib.rs
:
神经网络中常用数学函数的SIMD向量化实现
在此库中的每个函数都有多个变体,通常包括
- 一个操作标量的版本
- 一个从输入切片读取值并将其写入等长输出切片对应位置的版本。这些版本以
vec_
前缀命名。 - 一个从可变输入切片读取值并将计算结果就地写入的版本。这些版本以
vec_
前缀和_in_place
后缀命名。
所有变体都使用相同的底层实现,应该具有相同的精度。
有关精度的注释,请参阅源代码。