2 个版本
0.1.5-alpha.0 | 2023年3月25日 |
---|---|
0.1.4-alpha.0 | 2023年3月3日 |
#8 in #enforce
1MB
6K SLoC
caffe2op-enforce
caffe2op-enforce 是一个 Rust 包,它提供用于 DSP 和机器学习计算的数学运算符。该包包括 EnforceFinite 运算符,用于在计算期间强制 tensor 值的有限性。
注意:此包目前正在从 C++ 转译到 Rust,一些函数体可能仍在转译过程中。
EnforceFinite 运算符通常在数值稳定性很重要的情况下使用,例如在深度神经网络中。它接受一个 tensor 作为输入,并输出相同 tensor,其中所有非有限值被替换为指定的值。EnforceFinite 运算符可以帮助防止 NaN(非数字)值通过网络传播并导致数值不稳定性。
该包还包括 EnsureClipped 运算符,该运算符通过截断超出范围的值来确保 tensor 值在指定范围内。这也可以帮助防止神经网络中的数值不稳定性。
最后,还包括 CopyWithContext 运算符,以允许在不同上下文之间复制数据,例如在 CPU 和 GPU 内存之间。
总的来说,caffe2op-enforce 为机器学习和 DSP 计算提供了维护数值稳定性的有用运算符集。
依赖项
~36MB
~409K SLoC