本文内容:在不同位置添加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