2 个版本

0.1.5-alpha.02023年3月25日
0.1.4-alpha.02023年3月3日

#8 in #enforce

BSD-3-Clause

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