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))