Kolmogorov-Arnold Networks (KANs)_kolmogorov-arnold pytorch实现

KAN
这是个非常厉害的设计，看完论文我再写评价。

下面是实验，都是作者给出的代码

>>> from kan.KANLayer import KANLayer
>>> model = KANLayer(in_dim=3, out_dim=5)
>>> (model.in_dim, model.out_dim)
(3, 5)
1
2
3
4

>>> import torch
>>> x = torch.normal(0,1,size=(100,3))
>>> y, preacts, postacts, postspline = model(x)
>>> x = torch.normal(0,1,size=(4, 784, 1024))
>>> x = x.reshape(4*784, 1024)
>>> y, preacts, postacts, postspline = model2(x)
killed
1
2
3
4
5
6
7

由于计算量太大，直接嘎了。
所以寻求GPU的帮助是必要的，下面看一下论文以及代码再做说明！