import Playgrounds
import MLX

#Playground("Testing MLXArray") {
   let vector = MLXArray(converting: [9, 2, 3, 0, -2, 4])
   // array([9, 2, 3, 0, -2, 4], dtype=float32)
   
   let softmax = softmax(vector)
   // array([0.989836, 0.000902613, 0.00245356, 0.000122155, 1.65319e-05, 0.00666946], dtype=float32)
   
   let percentage = round(multiply(softmax, 100))
   // array([99, 0, 0, 0, 0, 1], dtype=float32)
}