Modifiy infer to acquire u/i bias and u/i matrix