Fix mypy errors in maths/sigmoid_linear_unit.py (#8786)

* updating DIRECTORY.md

* Fix mypy errors in sigmoid_linear_unit.py

* updating DIRECTORY.md

* updating DIRECTORY.md

---------

Co-authored-by: github-actions <${GITHUB_ACTOR}@users.noreply.github.com>
This commit is contained in:
Tianyi Zheng 2023-06-22 05:24:34 -07:00 committed by GitHub
parent 5b0890bd83
commit 5ffe601c86
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -17,7 +17,7 @@ This script is inspired by a corresponding research paper.
import numpy as np import numpy as np
def sigmoid(vector: np.array) -> np.array: def sigmoid(vector: np.ndarray) -> np.ndarray:
""" """
Mathematical function sigmoid takes a vector x of K real numbers as input and Mathematical function sigmoid takes a vector x of K real numbers as input and
returns 1/ (1 + e^-x). returns 1/ (1 + e^-x).
@ -29,17 +29,15 @@ def sigmoid(vector: np.array) -> np.array:
return 1 / (1 + np.exp(-vector)) return 1 / (1 + np.exp(-vector))
def sigmoid_linear_unit(vector: np.array) -> np.array: def sigmoid_linear_unit(vector: np.ndarray) -> np.ndarray:
""" """
Implements the Sigmoid Linear Unit (SiLU) or swish function Implements the Sigmoid Linear Unit (SiLU) or swish function
Parameters: Parameters:
vector (np.array): A numpy array consisting of real vector (np.ndarray): A numpy array consisting of real values
values.
Returns: Returns:
swish_vec (np.array): The input numpy array, after applying swish_vec (np.ndarray): The input numpy array, after applying swish
swish.
Examples: Examples:
>>> sigmoid_linear_unit(np.array([-1.0, 1.0, 2.0])) >>> sigmoid_linear_unit(np.array([-1.0, 1.0, 2.0]))