Fix mypy errors in maths/gaussian_error_linear_unit.py (#8610)

* updating DIRECTORY.md

* Fix mypy errors in gaussian_error_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-08-16 04:36:10 -07:00 committed by GitHub
parent 5c276a8377
commit beb43517c3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -30,12 +30,10 @@ def gaussian_error_linear_unit(vector: np.ndarray) -> np.ndarray:
Implements the Gaussian Error Linear Unit (GELU) function
Parameters:
vector (np.array): A numpy array of shape (1,n)
consisting of real values
vector (np.ndarray): A numpy array of shape (1, n) consisting of real values
Returns:
gelu_vec (np.array): The input numpy array, after applying
gelu.
gelu_vec (np.ndarray): The input numpy array, after applying gelu
Examples:
>>> gaussian_error_linear_unit(np.array([-1.0, 1.0, 2.0]))