Skip to content

Commit a9af92f

Browse files
Merge branch 'improve-softmax-typecheck' of https://github.com/shreyasNaik0101/Python into improve-softmax-typecheck
2 parents 388b0f9 + 10d6452 commit a9af92f

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

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)