Unet改进35：添加FastKANConv2DLayer(2024最新改进方法)

本文内容：在不同位置添加FastKANConv2DLayer

目录

论文简介

1.步骤一

2.步骤二

3.步骤三

4.步骤四

论文简介

地址

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