GPTCache

1
import numpy as np
2

3

4
def softmax(x: list):
5
    x = np.array(x)
6
    assert len(x.shape) == 1, f"Expect to get a shape of (len,) but got {x.shape}, x value: {x}."
7
    max_val = x.max()
8
    e_x = np.exp(x - max_val)
9
    return e_x / e_x.sum()
10