摘要: import torch import torch.nn as nn import torch.nn.functional as F class PCFN(nn.Module): ''' 使用带有GELU的激活函数的1*1卷积对扩展的隐藏空间进行跨信道交互。 然后将隐藏特征分割成两块 对其中一块使用 阅读全文
posted @ 2024-11-17 18:55 iceeci 阅读(7) 评论(0) 推荐(0) 编辑
摘要: import torch import torch.nn as nn import torch.nn.functional as F class DMlp(nn.Module): ''' 用来提取局部特征 ''' def __init__(self, dim, growth_rate=2.0): s 阅读全文
posted @ 2024-11-17 18:54 iceeci 阅读(3) 评论(0) 推荐(0) 编辑