out = torch.matmul(attn, v) ... return self.norm(out), attn class FeedForward(nn.Module): def __init__(self, dim, ... return x, attns
確定! 回上一頁