实验五、多层感知机的实现

第1关 实现全连接层的前向传播

import numpy as np

 

class FullyConnected:

    def __init__(self, W, b):

        r'''

        全连接层的初始化。

        Parameter:

        - W: numpy.array, (D_in, D_out)

        - b: numpy.array, (D_out)

        '''

        self.W = W

        self.b = b

        self.x = None

        self.original_x_shape = None

    def forward(self, x):

        r'''

        全连接层的前向传播。

        Parameter:

        - x: numpy.array, (B, d1, d2, ..., dk)

        Return:

        - y: numpy.array, (B, M)

        '''

        ########## Begin ##########

        return np.matmul(x,self.W)