Apache Arrow Tensor Extension的目标是为Apache Arrow数据格式提供对张量（Tensor）数据的支持。它允许在Arrow格式中存储和传输张量数据，以便在不同的计算框架和工具之间进行高效的数据交换。

以下是一个使用Apache Arrow Tensor Extension的代码示例：

import pyarrow as pa
import pyarrow.tensor as pt

# 创建一个2x3的浮点型张量
data = [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]]
tensor = pa.Tensor.from_numpy(data)

# 将张量转换为Arrow格式
tensor_array = pt.TensorArray.from_tensor(tensor)

# 将Arrow数组写入磁盘
with pa.RecordBatchFileWriter('tensor.arrow', tensor_array.schema) as writer:
    writer.write(tensor_array)

# 从磁盘读取Arrow数组
with pa.RecordBatchFileReader('tensor.arrow') as reader:
    result = reader.read_all()

# 将Arrow数组转换回张量
result_tensor = pt.Tensor.from_tensor_array(result)
print(result_tensor.to_numpy())

这个示例展示了如何创建一个张量，并将其转换为Arrow格式。然后，将Arrow数组写入磁盘并从磁盘读取回来。最后，将Arrow数组转换回张量，并打印出其值。