Convert Tensorflow Dataset To Numpy Array Multiplication In Java