Total Complexity | 2 |
Total Lines | 11 |
Duplicated Lines | 0 % |
1 | #!/usr/bin/env python |
||
7 | class Activation(NeuralLayer): |
||
8 | """ |
||
9 | Activation layer. |
||
10 | """ |
||
11 | |||
12 | def __init__(self, activation_type): |
||
13 | super(Activation, self).__init__(activation_type) |
||
14 | self._activation = build_activation(activation_type) |
||
15 | |||
16 | def compute_tensor(self, x): |
||
17 | return self._activation(x) |