本文内容:添加FastKANConv2DLayer
目录
论文简介
1.步骤一
新建blocks/fastkan.py文件,添加如下代码:
import torch
import torch.nn as nn
class RadialBasisFunction(nn.Module):
def __init__(
self,
grid_min: float = -2.,
grid_max: float = 2.,
num_grids: int = 8,
denominator: float = None, # larger denominators lead to smoother basis
):
super().__init__()
grid = torch.linspa