transformer结构输出部分

import torch
import torch.nn as nn
import torch.nn.functional as F

class generator(nn.Module):
        def __init__(self,d_model,vocab):
                super().__init__()

                self.linear=nn.Linear(d_model,vocab)

        def forward(self,input)
                return F.log_foftmax(self.linear(input),dim=-1) 

gen = Generator(d_model, vocab_size)
gen_result = gen(x)