| 1 |
def softmax(lst): |
| 2 |
norm = np.exp(lst - np.max(lst)) |
| 3 |
return norm / norm.sum() |
| 4 |
|
| 5 |
struct NDArray: |
| 6 |
def max(self) -> NDArray: |
| 7 |
return self.pmap(SIMD.max) |
| 8 |
|
| 9 |
struct SIMD[type: DType, width: Int]: |
| 10 |
def max(self, rhs: Self) -> Self: |
| 11 |
return (self >= rhs).select(self, rhs) |
