oreilly-japan / deep-learning-from-scratch

『ゼロから作る Deep Learning』(O'Reilly Japan, 2016)
MIT License
3.99k stars 3.34k forks source link

Update layers.py #35

Closed storypku closed 5 years ago

storypku commented 5 years ago

Suppose X.shape = [m, n, 3], W.shape = 3, 2. X should be reshaped to [m n, 3] rather than [m, 3 n].