attention代码
s计算score的方式有以下几种:import torch
class Attn(nn.module):def __init__(self, method ,hidden_size):super(Attn, self).__init__()self.method methodself.hidden_size hidden_sizeif self.method not in [dot, general, concat]:raise ValueError(self.…
2026/7/28 18:22:12