begining to better integrate batch with dataset/datatensor and models