Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- # dout: ∂L/∂o
- # a: the input of FC layer
- # the shape of a: (batch size, hidden layer neuron)
- # the shape of dw: (hidden layer neuron, num of classes)
- dw = a.reshape(a.shape[0], -1).T.dot(dout)
- db = np.sum(dout, axis=0)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement