import Playgrounds
import MLX

#Playground("Testing MLXArray") {
   let zeros = MLXArray.zeros([2, 3], type: Int16.self)
   // array([[0, 0, 0],
   //        [0, 0, 0]], dtype=int16)

   let ones = MLXArray.ones([2, 3])
   // array([[1, 1, 1],
   //        [1, 1, 1]], dtype=float32)

   let identity = MLXArray.eye(3, m: 3)
   // array([[1, 0, 0],
   //        [0, 1, 0],
   //        [0, 0, 1]], dtype=float32)

   let vector = MLXArray.linspace(Float32(0), Float32(1), count: 5)
   // array([0, 0.25, 0.5, 0.75, 1], dtype=float32)
}