Unbatch theoretical implementation, fix single batch jblas case.