File tree 1 file changed +4
-6
lines changed
1 file changed +4
-6
lines changed Original file line number Diff line number Diff line change 17
17
import numpy as np
18
18
19
19
20
- def sigmoid (vector : np .array ) -> np .array :
20
+ def sigmoid (vector : np .ndarray ) -> np .ndarray :
21
21
"""
22
22
Mathematical function sigmoid takes a vector x of K real numbers as input and
23
23
returns 1/ (1 + e^-x).
@@ -29,17 +29,15 @@ def sigmoid(vector: np.array) -> np.array:
29
29
return 1 / (1 + np .exp (- vector ))
30
30
31
31
32
- def sigmoid_linear_unit (vector : np .array ) -> np .array :
32
+ def sigmoid_linear_unit (vector : np .ndarray ) -> np .ndarray :
33
33
"""
34
34
Implements the Sigmoid Linear Unit (SiLU) or swish function
35
35
36
36
Parameters:
37
- vector (np.array): A numpy array consisting of real
38
- values.
37
+ vector (np.ndarray): A numpy array consisting of real values
39
38
40
39
Returns:
41
- swish_vec (np.array): The input numpy array, after applying
42
- swish.
40
+ swish_vec (np.ndarray): The input numpy array, after applying swish
43
41
44
42
Examples:
45
43
>>> sigmoid_linear_unit(np.array([-1.0, 1.0, 2.0]))
You can’t perform that action at this time.
0 commit comments