给我一段python float32转float16的代码

以下是一段 Python 中 float32 转 float16 的代码：

import numpy as np

def float32_to_float16(arr):
    return np.float16(arr).view(np.uint16)

# 示例
arr = np.array([1.0, 2.0, 3.0], dtype=np.float32)
arr_float16 = float32_to_float16(arr)
print(arr_float16)

这段代码使用了 NumPy 库中的 `float16` 类型和 `view` 方法，将 `float32` 类型的数组转换为 `float16` 类型的数组。