Chinese contentTorchCurves 发布:把参数化曲线变成 PyTorch 可学习单元
如果你在玩 KAN 或者函数回归,这个库挺有意思:曲线参数可以直接放进训练循环里一起优化,不用自己写 B 样条求值。作者还拿 EfficientKAN 做了性能对比,什么时候开销大会吃亏都说清楚了,写 PyTorch 自定义层的可以参考实现思路。
如果你在玩 KAN 或者函数回归,这个库挺有意思:曲线参数可以直接放进训练循环里一起优化,不用自己写 B 样条求值。作者还拿 EfficientKAN 做了性能对比,什么时候开销大会吃亏都说清楚了,写 PyTorch 自定义层的可以参考实现思路。