摘要: import torch import torch.nn as nn import torch.nn.functional as F class MultiHeadSelfAttention(nn.Module): def __init__(self, embed_size, heads): sup 阅读全文
posted @ 2024-02-26 11:31 15375357604 阅读(142) 评论(0) 推荐(0) 编辑