2025年3月19日
多头注意力机制运算过程详解
摘要： 记录一下用于学习多头注意力机制的计算过程的实验脚本 from TransUNet.networks.vit_seg_modeling import Attention import TransUNet.networks.vit_seg_configs as Config import torch, 阅读全文
posted @ 2025-03-19 12:17 yanyeeee 阅读(139) 评论(0) 推荐(0)