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

目录

论文简介

1.步骤一

2.步骤二

3.步骤三

4.步骤四

---

论文简介

地址

1.步骤一

新建blocks/kan_conv.py文件，添加如下代码：

from functools import lru_cache

import torch
import torch.nn as nn
from torch.nn.functional import conv3d, conv2d, conv1d
##来源：https://github.com/1ssb/torchkan

class KALNConvNDLayer(nn.Module):
    def __init__(self, conv_class, norm_class, conv_w_fun, input_dim, output_dim, degree, kernel_size,
                 groups=1, padding=0, stride=1, dilation=1, dropout: float = 0.0,
                 ndim