2025年3月21日
attention 代码
摘要： import torch import torch.nn as nn from math import sqrt class SelfAttention(nn.Module): def __init__(self, input_dim, dim_q_k, dim_v): self.q = nn.Li 阅读全文
posted @ 2025-03-21 14:17 15375357604 阅读(19) 评论(0) 推荐(0)