Function linear 

pub fn linear(x: &[f32], m: usize, k: usize, w: &[f32], n: usize) -> Vec<f32>

y = x @ w^T for x: [m, k], w: [n, k] (torch Linear layout), no bias.