CAFE

6. 딥러닝 수업

Re:문제80. (오늘의 마지막 문제) 책 175 페이지 아래에 있는 Affine 클래스를 생성하고 입력 전파를 흘려 보내시오 ~

작성자11기_이승혁|작성시간20.08.07|조회수24 목록 댓글 0

class Affine:

    def __init__(self,W,b):

        self.W=W

        self.b=b

        self.x=None

        self.dW=None

        self.db=None

 

    def forward(self,x):

        self.x=x

        out=np.dot(x,self.W)+self.b

        return out

 

    def backward(self,dout):

        dx=np.dot(dout,self.W.T)

        self.dW=np.dot(self.x.T,dout)

        self.db=np.sum(dout,aixs=0)

 

        return dx

 

x=np.array([[1,2],[3,4]]) # 2x2

W=np.array([[1,3,5],[2,4,6]]) # 2x3

b=np.array([[1,1,1]]) # 1x3

affine=Affine(W,b)

 

print(affine.forward(x))


다음검색
현재 게시글 추가 기능 열기

댓글

댓글 리스트
맨위로

카페 검색

카페 검색어 입력폼