【手撕算法系列】FFN

2023-12-17 15:49:41
class FFN(nn.Module):
    def __init__(self, in_ch, hid_ch, out_ch):
        super(FFN, self).__init__()
        self.fc1 = nn.Linear(in_ch, hid_ch)
        self.relu = nn.ReLU()
        self.fc2 = nn.Linear(hid_ch, out_ch)

    def forward(self, x):
        out = self.fc1(x)
        out = self.relu(out)
        out = self.fc2(out)
        return out

文章来源:https://blog.csdn.net/weixin_41845840/article/details/135044945
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。