import paddle
from paddle import nn
import numpy as np

class FeedForward(nn.Layer):
    def __init__(self, dim, hidden_dim, dropout=0.):

更多文章请关注《万象专栏》