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())
