def _to_ndarray(java_tensor: JavaObject) -> Any: if (java_tensor.getRank() == 0): return np.array(java_tensor.scalar()) else: return Tensor.__get_ndarray_from_tensor(java_tensor).reshape(java_tensor.getShape())
def _to_ndarray(java_tensor: JavaObject) -> numpy_types: if java_tensor.getRank() == 0: return np.array(java_tensor.scalar()) else: return np.array(list(java_tensor.asFlatArray())).reshape( java_tensor.getShape())