上海优化网站seo公司郑州经济技术开发区实验中学

本文内容：添加KACNConvNDLayer

目录

论文简介

1.步骤一

2.步骤二

3.步骤三

4.步骤四

论文简介

1.步骤一

新建block/kacn_conv.py文件，添加如下代码：

import torch
import torch.nn as nn##源码地址：https://github.com/SynodicMonth/ChebyKAN
class KACNConvNDLayer(nn.Module):def __init__(self, conv_class, norm_class, input_dim, output_dim, degree, kernel_size,groups=1, padding=0, stride=1, dilation=1,ndim: int = 2, dropout=0.0):super(KACNConvNDLayer, self).__init__()self.inputdim = input_dimself.outdim = output_dim