def test_dropout():
layer_test(core.Dropout,
kwargs={'p': 0.5},
input_shape=(3, 2))
layer_test(core.Dropout,
kwargs={'p': 0.5, 'noise_shape': [3, 1]},
input_shape=(3, 2))
layer_test(core.SpatialDropout1D,
kwargs={'p': 0.5},
input_shape=(2, 3, 4))
layer_test(core.SpatialDropout2D,
kwargs={'p': 0.5},
input_shape=(2, 3, 4, 5))
layer_test(core.SpatialDropout3D,
kwargs={'p': 0.5},
input_shape=(2, 3, 4, 5, 6))
评论列表
文章目录