Skip to content

Commit 10d6452

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 8e09162 commit 10d6452

1 file changed

Lines changed: 1 addition & 4 deletions

File tree

maths/softmax.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,7 @@
22
from typing import Optional
33

44

5-
def softmax(
6-
vector: np.ndarray | list | tuple,
7-
axis: int | None = -1
8-
) -> np.ndarray:
5+
def softmax(vector: np.ndarray | list | tuple, axis: int | None = -1) -> np.ndarray:
96
"""
107
Compute the softmax of `vector` along `axis` in a numerically-stable way.
118

0 commit comments

Comments
 (0)