Skip to content

Commit 5ee7dcb

Browse files
committed
add unit tests
1 parent 560ad77 commit 5ee7dcb

File tree

1 file changed

+152
-0
lines changed

1 file changed

+152
-0
lines changed

python/paddle/fluid/tests/unittests/test_layers.py

Lines changed: 152 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -573,6 +573,158 @@ def test_prelu(self):
573573
self.assertIsNotNone(out)
574574
print(str(program))
575575

576+
def test_brelu(self):
577+
program = Program()
578+
with program_guard(program):
579+
input = layers.data(name="input", shape=[16], dtype="float32")
580+
out = layers.brelu(input, t_min=1.0, t_max=20.0, name='brelu')
581+
self.assertIsNotNone(out)
582+
print(str(program))
583+
584+
def test_leaky_relu(self):
585+
program = Program()
586+
with program_guard(program):
587+
input = layers.data(name="input", shape=[16], dtype="float32")
588+
out = layers.leaky_relu(input, alpha=0.1, name='leaky_relu')
589+
self.assertIsNotNone(out)
590+
print(str(program))
591+
592+
def test_soft_relu(self):
593+
program = Program()
594+
with program_guard(program):
595+
input = layers.data(name="input", shape=[16], dtype="float32")
596+
out = layers.soft_relu(input, threshold=30.0, name='soft_relu')
597+
self.assertIsNotNone(out)
598+
print(str(program))
599+
600+
def test_sigmoid(self):
601+
program = Program()
602+
with program_guard(program):
603+
input = layers.data(name="input", shape=[16], dtype="float32")
604+
out = layers.sigmoid(input, name='sigmoid')
605+
self.assertIsNotNone(out)
606+
print(str(program))
607+
608+
def test_logsigmoid(self):
609+
program = Program()
610+
with program_guard(program):
611+
input = layers.data(name="input", shape=[16], dtype="float32")
612+
out = layers.logsigmoid(input, name='logsigmoid')
613+
self.assertIsNotNone(out)
614+
print(str(program))
615+
616+
def test_exp(self):
617+
program = Program()
618+
with program_guard(program):
619+
input = layers.data(name="input", shape=[16], dtype="float32")
620+
out = layers.exp(input, name='exp')
621+
self.assertIsNotNone(out)
622+
print(str(program))
623+
624+
def test_tanh(self):
625+
program = Program()
626+
with program_guard(program):
627+
input = layers.data(name="input", shape=[16], dtype="float32")
628+
out = layers.tanh(input, name='tanh')
629+
self.assertIsNotNone(out)
630+
print(str(program))
631+
632+
def test_tanh_shrink(self):
633+
program = Program()
634+
with program_guard(program):
635+
input = layers.data(name="input", shape=[16], dtype="float32")
636+
out = layers.tanh_shrink(input, name='tanh_shrink')
637+
self.assertIsNotNone(out)
638+
print(str(program))
639+
640+
def test_sqrt(self):
641+
program = Program()
642+
with program_guard(program):
643+
input = layers.data(name="input", shape=[16], dtype="float32")
644+
out = layers.sqrt(input, name='sqrt')
645+
self.assertIsNotNone(out)
646+
print(str(program))
647+
648+
def test_abs(self):
649+
program = Program()
650+
with program_guard(program):
651+
input = layers.data(name="input", shape=[16], dtype="float32")
652+
out = layers.abs(input, name='abs')
653+
self.assertIsNotNone(out)
654+
print(str(program))
655+
656+
def test_ceil(self):
657+
program = Program()
658+
with program_guard(program):
659+
input = layers.data(name="input", shape=[16], dtype="float32")
660+
out = layers.ceil(input, name='ceil')
661+
self.assertIsNotNone(out)
662+
print(str(program))
663+
664+
def test_floor(self):
665+
program = Program()
666+
with program_guard(program):
667+
input = layers.data(name="input", shape=[16], dtype="float32")
668+
out = layers.floor(input, name='floor')
669+
self.assertIsNotNone(out)
670+
print(str(program))
671+
672+
def test_cos(self):
673+
program = Program()
674+
with program_guard(program):
675+
input = layers.data(name="input", shape=[16], dtype="float32")
676+
out = layers.cos(input, name='cos')
677+
self.assertIsNotNone(out)
678+
print(str(program))
679+
680+
def test_sin(self):
681+
program = Program()
682+
with program_guard(program):
683+
input = layers.data(name="input", shape=[16], dtype="float32")
684+
out = layers.sin(input, name='sin')
685+
self.assertIsNotNone(out)
686+
print(str(program))
687+
688+
def test_round(self):
689+
program = Program()
690+
with program_guard(program):
691+
input = layers.data(name="input", shape=[16], dtype="float32")
692+
out = layers.round(input, name='round')
693+
self.assertIsNotNone(out)
694+
print(str(program))
695+
696+
def test_reciprocal(self):
697+
program = Program()
698+
with program_guard(program):
699+
input = layers.data(name="input", shape=[16], dtype="float32")
700+
out = layers.reciprocal(input, name='reciprocal')
701+
self.assertIsNotNone(out)
702+
print(str(program))
703+
704+
def test_square(self):
705+
program = Program()
706+
with program_guard(program):
707+
input = layers.data(name="input", shape=[16], dtype="float32")
708+
out = layers.square(input, name='square')
709+
self.assertIsNotNone(out)
710+
print(str(program))
711+
712+
def test_softplus(self):
713+
program = Program()
714+
with program_guard(program):
715+
input = layers.data(name="input", shape=[16], dtype="float32")
716+
out = layers.softplus(input, name='softplus')
717+
self.assertIsNotNone(out)
718+
print(str(program))
719+
720+
def test_softsign(self):
721+
program = Program()
722+
with program_guard(program):
723+
input = layers.data(name="input", shape=[16], dtype="float32")
724+
out = layers.softsign(input, name='softsign')
725+
self.assertIsNotNone(out)
726+
print(str(program))
727+
576728
def test_sequence_enumerate(self):
577729
program = Program()
578730
with program_guard(program):

0 commit comments

Comments
 (0)