let (*) (a: float [,]) (b: float [,]) =
    let rows = a.GetLength 0
    let cols = b.GetLength 1
    Array2D.init rows cols (fun x y -> inner (flatten a.[x..x,0..]) (flatten b.[0..,y..y]))