let log_grad x y y' =   let open MX in   let y' = sigmoid y' in   (transpose x) $@ (y' -@ y) /$ (float_of_int (row_num x))