Add the test_repeat_layer.py

f4853510 · xuwei06 · 226f8103 · f4853510
隐藏空白更改
内联并排

Showing with 11 addition and 0 deletion

python/paddle/trainer_config_helpers/tests/configs/test_repeat_layer.py ...trainer_config_helpers/tests/configs/test_repeat_layer.py +11 -0

未找到文件。
--- a/python/paddle/trainer_config_helpers/tests/configs/test_repeat_layer.py
+++ b/python/paddle/trainer_config_helpers/tests/configs/test_repeat_layer.py
+from paddle.trainer_config_helpers import *
+
+settings(batch_size=1000, learning_rate=1e-5)
+
+din = data_layer(name='data', size=30)
+
+outputs(
+    repeat_layer(
+        input=din, num_repeats=10, as_row_vector=True),
+    repeat_layer(
+        input=din, num_repeats=10, act=TanhActivation(), as_row_vector=False))