def test_Dropout():
from npdl.layers import Dropout
input = np.random.rand(10, 20)
pre_grad = np.random.rand(10, 20)
layer = Dropout(0.5)
layer.connect_to(PrevLayer((10, 20)))
assert layer.forward(input).shape == input.shape
assert np.allclose(layer.forward(input, False), input * 0.5)
assert layer.backward(pre_grad).shape == input.shape
layer = Dropout()
layer.connect_to(PrevLayer((10, 20)))
assert np.allclose(layer.forward(input), input)
assert layer.backward(pre_grad).shape == input.shape
评论列表
文章目录